diff --git a/src/DynamicModel.cc b/src/DynamicModel.cc
index 9dfec6b170f9f21a44e1feedd63eb8197f2a9479..192e436bab30586d06002a1f94ef3cb7b631bd66 100644
--- a/src/DynamicModel.cc
+++ b/src/DynamicModel.cc
@@ -4546,19 +4546,17 @@ DynamicModel::ParamUsedWithLeadLag() const
 }
 
 void
-DynamicModel::createVariableMapping(int orig_eq_nbr)
+DynamicModel::createVariableMapping()
 {
-  for (int ii = 0; ii < orig_eq_nbr; ii++)
+  for (size_t ii = 0; ii < equations.size(); ii++)
     {
       set<int> eqvars;
       equations[ii]->collectVariables(SymbolType::endogenous, eqvars);
       equations[ii]->collectVariables(SymbolType::exogenous, eqvars);
       for (auto eqvar : eqvars)
-        {
-          eqvar = symbol_table.getUltimateOrigSymbID(eqvar);
-          if (eqvar >= 0 && !symbol_table.isAuxiliaryVariable(eqvar))
-            variableMapping[eqvar].emplace(ii);
-        }
+        if (int orig_symb_id = symbol_table.getUltimateOrigSymbID(eqvar);
+            orig_symb_id >= 0)
+        variableMapping[orig_symb_id].emplace(ii);
     }
 }
 
diff --git a/src/DynamicModel.hh b/src/DynamicModel.hh
index 7f0044752d3daa60c43e5cd3f2d8e016edb10414..8f7e2dacaf258bbdc5eed0171ad93e02b219fce0 100644
--- a/src/DynamicModel.hh
+++ b/src/DynamicModel.hh
@@ -396,7 +396,7 @@ public:
   void writeDynamicJacobianNonZeroElts(const string &basename) const;
 
   //! Creates mapping for variables and equations they are present in
-  void createVariableMapping(int orig_eq_nbr);
+  void createVariableMapping();
 
   //! Expands equation tags with default equation names (available "name" tag or LHS variable or equation ID)
   void expandEqTags();
diff --git a/src/ModFile.cc b/src/ModFile.cc
index 8c4f43b85375df5fdb1046b1dff7121c4246811d..902f05634034cba4e1e67061249d4427d08ac32c 100644
--- a/src/ModFile.cc
+++ b/src/ModFile.cc
@@ -559,9 +559,7 @@ ModFile::transformPass(bool nostrict, bool stochastic, bool compute_xrefs, bool
     }
   // And finally perform the substitutions
   dynamic_model.substituteVarExpectation(var_expectation_subst_table);
-  dynamic_model.createVariableMapping(mod_file_struct.orig_eq_nbr +
-                                      (mod_file_struct.ramsey_model_present ?
-                                       mod_file_struct.ramsey_eq_nbr : 0));
+  dynamic_model.createVariableMapping();
 
   /* Create auxiliary vars for leads and lags greater than 2, on both endos and
      exos. The transformation is not exactly the same on stochastic and