diff --git a/src/brevitas/graph/calibrate.py b/src/brevitas/graph/calibrate.py
index 0e1920ef9..bb435b7ef 100644
--- a/src/brevitas/graph/calibrate.py
+++ b/src/brevitas/graph/calibrate.py
@@ -93,9 +93,9 @@ class load_quant_model:
 
     def __init__(self, model):
         self.model = model
+        self.tracked_modules = []
 
     def __enter__(self):
-        self.tracked_modules = []
         for module in self.model.modules():
             if issubclass(type(module), QuantWBIOL):
                 if module.bias is None: