diff --git a/agent.py b/agent.py
index dd60c439f0397d1d6cfd43635a4d1220a9ea16e4..2a1c391556537021433ffaa2fc2d5a8bed71bc42 100644
--- a/agent.py
+++ b/agent.py
@@ -84,7 +84,8 @@ if __name__ == '__main__':
     infos = [{} for _ in range(num_envs)]
 
     # Change this to your agent interface
-    num_actions = batched_env.envs[0].action_space.n
+    # num_actions = batched_env.envs[0].action_space.n
+    num_actions = 112
     agent = RandomBatchedAgent(num_envs, num_actions)
     
     # The evaluation setup will automatically stop after the requisite number of rollouts