diff --git a/train_rl_with_realtime.py b/train_rl_with_realtime.py
index 5d67ddb..078a883 100644
--- a/train_rl_with_realtime.py
+++ b/train_rl_with_realtime.py
@@ -534,7 +534,7 @@ def run_training_thread(chart):
     def training_thread_func():
         try:
             # Use a small number of episodes to test termination handling
-            integrator.start_training(num_episodes=100, max_steps=2000)
+            integrator.start_training(num_episodes=100, max_steps=1000)
         except Exception as e:
             logger.error(f"Error in training thread: {str(e)}")