diff --git a/tests/conftest.py b/tests/conftest.py
index 740f65b7..41c1bc94 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -182,8 +182,8 @@ def run_generic(env, config_values):
             # Send the observation space to the agent to get an action
             # TEMP - random action for now
             # action = env.blue_agent_action(obs)
-            # action = env.action_space.sample()
-            action = 0
+            # action = 0
+            action = env.action_space.sample()
             # Run the simulation step on the live environment
             obs, reward, done, info = env.step(action)