diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py
index b7d5dbaa7..4ab31ea80 100644
--- a/src/brevitas/graph/equalize.py
+++ b/src/brevitas/graph/equalize.py
@@ -1300,6 +1300,8 @@ def apply(self, alpha):
                     co_optimize_act_weights=self.co_optimize_act_weights)
             scale_factors.append(scale_factor_region)
             rewriters.append(rewriters_region)
+            for r in rewriters_region:
+                r.apply(self.model)
         return scale_factors, rewriters_region
 
     def insert_mul_node(self, scale, shape, axis, act_node, batch_dim=0):