diff --git a/src/evox/workflows/std_workflow.py b/src/evox/workflows/std_workflow.py
index 848f48f9..c2dba4d9 100644
--- a/src/evox/workflows/std_workflow.py
+++ b/src/evox/workflows/std_workflow.py
@@ -252,7 +252,7 @@ def _step(self, state):
             for monitor in self.registered_hooks["post_step"]:
                 monitor.post_step(state)
 
-            return train_info, state
+            return state
 
         if self.jit_step:
             # the first argument is self, which should be static