diff --git a/src/lava/magma/core/callback_fx.py b/src/lava/magma/core/callback_fx.py
index ed635b896..1b1195f3e 100644
--- a/src/lava/magma/core/callback_fx.py
+++ b/src/lava/magma/core/callback_fx.py
@@ -73,14 +73,14 @@ def __init__(self,
 
     def pre_run_callback(self,
                          board: NxBoard = None,
-                         _var_id_to_var_model_map: dict = None
+                         var_id_to_var_model_map: dict = None
                          ) -> None:
         for fx in self.pre_run_fxs:
             fx(board)
 
     def post_run_callback(self,
                           board: NxBoard = None,
-                          _var_id_to_var_model_map: dict = None
+                          var_id_to_var_model_map: dict = None
                           ) -> None:
         for fx in self.post_run_fxs:
             fx(board)