1 files changed, 3 insertions, 9 deletions
diff --git a/src/theory/theory_state.cpp b/src/theory/theory_state.cpp
index e3655e3ab..ec7955125 100644
--- a/src/theory/theory_state.cpp
+++ b/src/theory/theory_state.cpp
@@ -21,23 +21,17 @@ namespace cvc5 {
 namespace theory {
 
 TheoryState::TheoryState(Env& env, Valuation val)
-    : EnvObj(env),
-      d_valuation(val),
-      d_ee(nullptr),
-      d_conflict(d_env.getContext(), false)
+    : EnvObj(env), d_valuation(val), d_ee(nullptr), d_conflict(context(), false)
 {
 }
 
 void TheoryState::setEqualityEngine(eq::EqualityEngine* ee) { d_ee = ee; }
 
-context::Context* TheoryState::getSatContext() const
-{
-  return d_env.getContext();
-}
+context::Context* TheoryState::getSatContext() const { return context(); }
 
 context::UserContext* TheoryState::getUserContext() const
 {
-  return d_env.getUserContext();
+  return userContext();
 }
 
 bool TheoryState::hasTerm(TNode a) const