diff --git a/domainlab/algos/trainers/fbopt_mu_controller.py b/domainlab/algos/trainers/fbopt_mu_controller.py
index 3193ae225..2429a3bd6 100644
--- a/domainlab/algos/trainers/fbopt_mu_controller.py
+++ b/domainlab/algos/trainers/fbopt_mu_controller.py
@@ -252,3 +252,6 @@ def update_setpoint(self, epo_reg_loss, epo_task_loss):
         update setpoint
         """
         return self.set_point_controller.observe(epo_reg_loss, epo_task_loss)
+
+    def __call__(self, epoch):
+        return self.mmu