1 files changed, 14 insertions, 13 deletions
diff --git a/src/theory/arith/dio_solver.cpp b/src/theory/arith/dio_solver.cpp
index f8b8e0e7b..71ad6de45 100644
--- a/src/theory/arith/dio_solver.cpp
+++ b/src/theory/arith/dio_solver.cpp
@@ -13,12 +13,13 @@
  **
  ** A Diophantine equation solver for the theory of arithmetic.
  **/
+#include "theory/arith/dio_solver.h"
 
 #include <iostream>
 
 #include "base/output.h"
 #include "options/arith_options.h"
-#include "theory/arith/dio_solver.h"
+#include "smt/smt_statistics_registry.h"
 
 using namespace std;
 
@@ -56,25 +57,25 @@ DioSolver::Statistics::Statistics() :
   d_conflictTimer("theory::arith::dio::conflictTimer"),
   d_cutTimer("theory::arith::dio::cutTimer")
 {
-  StatisticsRegistry::registerStat(&d_conflictCalls);
-  StatisticsRegistry::registerStat(&d_cutCalls);
+  smtStatisticsRegistry()->registerStat(&d_conflictCalls);
+  smtStatisticsRegistry()->registerStat(&d_cutCalls);
 
-  StatisticsRegistry::registerStat(&d_cuts);
-  StatisticsRegistry::registerStat(&d_conflicts);
+  smtStatisticsRegistry()->registerStat(&d_cuts);
+  smtStatisticsRegistry()->registerStat(&d_conflicts);
 
-  StatisticsRegistry::registerStat(&d_conflictTimer);
-  StatisticsRegistry::registerStat(&d_cutTimer);
+  smtStatisticsRegistry()->registerStat(&d_conflictTimer);
+  smtStatisticsRegistry()->registerStat(&d_cutTimer);
 }
 
 DioSolver::Statistics::~Statistics(){
-  StatisticsRegistry::unregisterStat(&d_conflictCalls);
-  StatisticsRegistry::unregisterStat(&d_cutCalls);
+  smtStatisticsRegistry()->unregisterStat(&d_conflictCalls);
+  smtStatisticsRegistry()->unregisterStat(&d_cutCalls);
 
-  StatisticsRegistry::unregisterStat(&d_cuts);
-  StatisticsRegistry::unregisterStat(&d_conflicts);
+  smtStatisticsRegistry()->unregisterStat(&d_cuts);
+  smtStatisticsRegistry()->unregisterStat(&d_conflicts);
 
-  StatisticsRegistry::unregisterStat(&d_conflictTimer);
-  StatisticsRegistry::unregisterStat(&d_cutTimer);
+  smtStatisticsRegistry()->unregisterStat(&d_conflictTimer);
+  smtStatisticsRegistry()->unregisterStat(&d_cutTimer);
 }
 
 bool DioSolver::queueConditions(TrailIndex t){