diff --git a/maro/rl/training/trajectory.py b/maro/rl/training/trajectory.py
index 5a014c698..cefac38e8 100644
--- a/maro/rl/training/trajectory.py
+++ b/maro/rl/training/trajectory.py
@@ -18,7 +18,7 @@ def get_action(self, action_by_agent, event) -> dict:
     def get_reward(self) -> float:
         pass
 
-    def on_env_feedback(self):
+    def on_env_feedback(self, event, state_by_agent, action_by_agent, reward):
         pass
 
     def on_finish(self):