diff --git a/rocketpy/control/controller.py b/rocketpy/control/controller.py
index d67ef5282..a68de1a28 100644
--- a/rocketpy/control/controller.py
+++ b/rocketpy/control/controller.py
@@ -84,6 +84,7 @@ def __init__(
         None
         """
         self.interactive_objects = interactive_objects
+        self.base_controller_function = controller_function
         self.controller_function = self.__init_controller_function(controller_function)
         self.sampling_rate = sampling_rate
         self.initial_observed_variables = initial_observed_variables