diff --git a/src/brevitas/export/onnx/debug.py b/src/brevitas/export/onnx/debug.py
index 85acba31b..9f669c43c 100644
--- a/src/brevitas/export/onnx/debug.py
+++ b/src/brevitas/export/onnx/debug.py
@@ -51,7 +51,5 @@ def enable_debug(model, input_enabled=True, output_enabled=True, proxy_level=Fal
     for name, module in model.named_modules():
         if hasattr(module, "export_debug_name") and filter_fn(module):
             module.export_debug_name = name
-            module.export_input_debug = input_enabled
-            module.export_output_debug = output_enabled
             module.register_forward_hook(hook)
     return hook