diff --git a/overcooked_simulator/gui_2d_vis/overcooked_gui.py b/overcooked_simulator/gui_2d_vis/overcooked_gui.py
index 62e5ca6335f20c1c0d6ec49b1e815cde06e4e6d5..7e280cf038b5812086724f288f16874f1197d7c5 100644
--- a/overcooked_simulator/gui_2d_vis/overcooked_gui.py
+++ b/overcooked_simulator/gui_2d_vis/overcooked_gui.py
@@ -275,7 +275,7 @@ class PyGameGUI:
         # Axis 0: joy stick left: -1 = left, ~0 = center, 1 = right
         # Axis 1: joy stick left: -1 = up, ~0 = center, 1 = down
         # see control stuff here (at the end of the page): https://www.pygame.org/docs/ref/joystick.html
-        for key_set in self.player_key_sets:
+        for key_set in self.key_sets:
             # if a joystick is connected for current player
             if joysticks[key_set.joystick]:
                 # Usually axis run in pairs, up/down for one, and left/right for the other. Triggers count as axes.
@@ -343,7 +343,7 @@ class PyGameGUI:
             joysticks: list of joysticks
         """
 
-        for key_set in self.player_key_sets:
+        for key_set in self.key_sets:
             # if a joystick is connected for current player
             if joysticks[key_set.joystick]:
                 # pickup = Button A <-> 0