1 files changed, 5 insertions, 1 deletions
diff --git a/src/theory/theory.cpp b/src/theory/theory.cpp
index 9669d97e0..7220e2e1c 100644
--- a/src/theory/theory.cpp
+++ b/src/theory/theory.cpp
@@ -104,6 +104,10 @@ void Theory::setEqualityEngine(eq::EqualityEngine* ee)
 {
   // set the equality engine pointer
   d_equalityEngine = ee;
+  if (d_theoryState != nullptr)
+  {
+    d_theoryState->setEqualityEngine(ee);
+  }
 }
 void Theory::setQuantifiersEngine(QuantifiersEngine* qe)
 {
@@ -127,7 +131,7 @@ void Theory::finishInitStandalone()
     d_allocEqualityEngine.reset(new eq::EqualityEngine(
         *esi.d_notify, d_satContext, esi.d_name, esi.d_constantsAreTriggers));
     // use it as the official equality engine
-    d_equalityEngine = d_allocEqualityEngine.get();
+    setEqualityEngine(d_allocEqualityEngine.get());
   }
   finishInit();
 }