diff --git a/src/gamemodes.py b/src/gamemodes.py
index 6b53942..d007854 100644
--- a/src/gamemodes.py
+++ b/src/gamemodes.py
@@ -104,7 +104,7 @@ class GameMode:
         pass
 
     # Here so any game mode can use it
-    def lovers_chk_win(self, evt, cli, var, rolemap, lpl, lwolves, lrealwolves):
+    def lovers_chk_win(self, evt, cli, var, rolemap, mainroles, lpl, lwolves, lrealwolves):
         winner = evt.data["winner"]
         if winner is not None and winner.startswith("@"):
             return # fool won, lovers can't win even if they would
@@ -119,7 +119,7 @@ class GameMode:
             evt.data["additional_winners"] = list(lovers)
             evt.data["message"] = messages["lovers_win"]
 
-    def all_dead_chk_win(self, evt, cli, var, rolemap, lpl, lwolves, lrealwolves):
+    def all_dead_chk_win(self, evt, cli, var, rolemap, mainroles, lpl, lwolves, lrealwolves):
         if evt.data["winner"] == "no_team_wins":
             evt.data["winner"] = "everyone"
             evt.data["message"] = messages["everyone_died_won"]
@@ -198,7 +198,7 @@ class VillagergameMode(GameMode):
         events.remove_listener("transition_day_begin", self.transition_day)
         events.remove_listener("retribution_kill", self.on_retribution_kill, priority=4)
 
-    def chk_win(self, evt, cli, var, rolemap, lpl, lwolves, lrealwolves):
+    def chk_win(self, evt, cli, var, rolemap, mainroles, lpl, lwolves, lrealwolves):
         # village can only win via unanimous vote on the bot nick
         # villagergame_lose should probably explain that mechanic
         # Note: not implemented here since that needs to work in default too