diff --git a/pytron_run/ai_manager/dummy/ai.py b/pytron_run/ai_manager/dummy/ai.py
index c6eacd5..9cdba62 100644
--- a/pytron_run/ai_manager/dummy/ai.py
+++ b/pytron_run/ai_manager/dummy/ai.py
@@ -7,5 +7,5 @@ class Ai(Player):
     def __init__(self):
         super(Ai, self).__init__()
 
-    def action(self, game):
+    def action(self, map, id):
         return Direction.RIGHT
diff --git a/pytron_run/ai_manager/dummy2/ai.py b/pytron_run/ai_manager/dummy2/ai.py
index e15e769..60d20be 100644
--- a/pytron_run/ai_manager/dummy2/ai.py
+++ b/pytron_run/ai_manager/dummy2/ai.py
@@ -4,5 +4,5 @@ class Ai(Player):
     def __init__(self):
         super(Ai, self).__init__()
 
-    def action(self, game):
+    def action(self, map, id):
         return Direction.LEFT