diff --git a/agent_zoo/demo_pong.py b/agent_zoo/demo_pong.py
index aadf9d3..014fe25 100644
--- a/agent_zoo/demo_pong.py
+++ b/agent_zoo/demo_pong.py
@@ -37,5 +37,6 @@ def play(env, pi, video):
         pi = Pol1(env.observation_space, env.action_space)
     else:
         pi = Pol2(env.observation_space, env.action_space)
-    play(env, pi, video=False)   # set video = player_n==0 to record video
+    while True:
+        play(env, pi, video=False)   # set video = player_n==0 to record video