diff --git a/src/omlt/linear_tree/lt_formulation.py b/src/omlt/linear_tree/lt_formulation.py
index 8ecac671..2d313fc3 100644
--- a/src/omlt/linear_tree/lt_formulation.py
+++ b/src/omlt/linear_tree/lt_formulation.py
@@ -106,6 +106,7 @@ def _build_formulation(self):
             output_vars=self.block.scaled_outputs,
             transformation=self.transformation,
             epsilon=self.epsilon,
+            include_leaf_equalities=True
         )
 
 
@@ -266,7 +267,7 @@ def _add_gdp_formulation_to_block(  # noqa: PLR0913
     output_vars,
     transformation,
     epsilon,
-    include_leaf_equalities=True,
+    include_leaf_equalities,
 ):
     """This function adds the GDP representation to the OmltBlock using Pyomo.GDP.