1 files changed, 5 insertions, 2 deletions
diff --git a/src/theory/model_manager_distributed.cpp b/src/theory/model_manager_distributed.cpp
index f3a501f94..4124407be 100644
--- a/src/theory/model_manager_distributed.cpp
+++ b/src/theory/model_manager_distributed.cpp
@@ -15,6 +15,7 @@
 
 #include "theory/model_manager_distributed.h"
 
+#include "smt/env.h"
 #include "theory/theory_engine.h"
 #include "theory/theory_model.h"
 #include "theory/theory_model_builder.h"
@@ -23,8 +24,9 @@ namespace cvc5 {
 namespace theory {
 
 ModelManagerDistributed::ModelManagerDistributed(TheoryEngine& te,
+                                                 Env& env,
                                                  EqEngineManager& eem)
-    : ModelManager(te, eem)
+    : ModelManager(te, env, eem)
 {
 }
 
@@ -69,10 +71,11 @@ bool ModelManagerDistributed::prepareModel()
   // model, which includes both dump their equality information and assigning
   // values. Notice the order of theories here is important and is the same
   // as the list in CVC5_FOR_EACH_THEORY in theory_engine.cpp.
+  const LogicInfo& logicInfo = d_env.getLogicInfo();
   for (TheoryId theoryId = theory::THEORY_FIRST; theoryId < theory::THEORY_LAST;
        ++theoryId)
   {
-    if (!d_logicInfo.isTheoryEnabled(theoryId))
+    if (!logicInfo.isTheoryEnabled(theoryId))
     {
       // theory not active, skip
       continue;