diff --git a/rsl_rl/rsl_rl/algorithms/him_ppo.py b/rsl_rl/rsl_rl/algorithms/him_ppo.py
index 42f1bdd..02d83c3 100644
--- a/rsl_rl/rsl_rl/algorithms/him_ppo.py
+++ b/rsl_rl/rsl_rl/algorithms/him_ppo.py
@@ -189,4 +189,4 @@ def update(self):
         mean_swap_loss /= num_updates
         self.storage.clear()
 
-        return mean_value_loss, mean_surrogate_loss, estimation_loss, swap_loss
+        return mean_value_loss, mean_surrogate_loss, mean_estimation_loss, mean_swap_loss