diff --git a/breakthebank.py b/breakthebank.py
index 68e6af8..4196b2c 100644
--- a/breakthebank.py
+++ b/breakthebank.py
@@ -118,8 +118,8 @@ def drawStageSelection(self):
                 if (stage_placeholderbutton_1.isOver(mouse) and 1<=levels_to_draw):
                     button_hover.play()
                     print("TRIGGERED stage selection -> in game")
-                    self.current_level = level_map_0
-                    self.current_level_parems = level0_param
+                    self.current_level = level_map
+                    self.current_level_parems = level_param
                     self.level = Level(self.current_level, self.current_level_parems, screen, "./imgs/stage1_lobby.png")
                     self.in_game = True
                 if (stage_placeholderbutton_2.isOver(mouse) and 2<=levels_to_draw):
@@ -205,7 +205,6 @@ def drawPauseMenu(self):
                     self.select_stage = True
                 if restart_button.isOver(mouse):
                     self.level = Level(self.current_level, self.current_level_parems, screen, "./imgs/stage1_lobby.png")
-                    self.level = Level(self.current_level, level0_param, screen, "./imgs/stage1_lobby.png")
                     self.in_game = True
                     self.musicON = False
             if event.type == pygame.KEYDOWN: