diff --git a/src/extra.py b/src/extra.py
index 5b6af80916d513619b7b9b3c1edbd18c0e24ee04..b70830f91714c3a8a3efdabe34bd1a427ea1f73e 100644
--- a/src/extra.py
+++ b/src/extra.py
@@ -352,7 +352,7 @@ class Extra(ObservationBuilder):
         return action_dict
 
     def rl_agent_act(self, observation, info, eps=0.0):
-        if len(self.random_agent_starter) != len(self.env.get_num_agents()):
+        if len(self.random_agent_starter) != self.env.get_num_agents():
             self.random_agent_starter = np.random.random(self.env.get_num_agents()) * 1000.0
             self.loadAgent()