diff --git a/src/extra.py b/src/extra.py
index f9f015bbdb27bb0ec8c064811ad2c27ffe54a93d..312cebb18d12f36da35a4f8af13d10d597d89f1b 100644
--- a/src/extra.py
+++ b/src/extra.py
@@ -339,7 +339,7 @@ class Extra(ObservationBuilder):
 
         return observation
 
-    def rl_agent_act(self, observation, info, eps=0.0):
+    def rl_agent_act_ADRIAN(self, observation, info, eps=0.0):
         self.loadAgent()
         action_dict = {}
         for a in range(self.env.get_num_agents()):
@@ -351,7 +351,7 @@ class Extra(ObservationBuilder):
 
         return action_dict
 
-    def rl_agent_act_ADRIAN(self, observation, info, eps=0.0):
+    def rl_agent_act(self, observation, info, eps=0.0):
         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()