diff --git a/source/core/src/main/com/csse3200/game/screens/MainGameScreen.java b/source/core/src/main/com/csse3200/game/screens/MainGameScreen.java
index c1c1459d..f11a4aa0 100644
--- a/source/core/src/main/com/csse3200/game/screens/MainGameScreen.java
+++ b/source/core/src/main/com/csse3200/game/screens/MainGameScreen.java
@@ -213,14 +213,11 @@ public void render(float delta) {
     ServiceLocator.getWaveService().getDisplay().updateMobCount();
     renderer.render();
 
-    // Check if the game has ended
     // Check if the game has ended
     if (ServiceLocator.getGameEndService().hasGameEnded()) {
       ui.getEvents().trigger("lose");
-    }
-
-    // Check if all waves are completed and the level has been completed
-    if (ServiceLocator.getWaveService().isLevelCompleted()) {
+    } else if (ServiceLocator.getWaveService().isLevelCompleted()) {
+      // Check if all waves are completed and the level has been completed
       if (selectedLevel == 2) { // Lava level
         // If it's the lava level, go to the "win" screen
         ui.getEvents().trigger("win");
@@ -232,7 +229,6 @@ public void render(float delta) {
       }
     }
   }
-  // Add something in to unlock the next planet/level?
 
   @Override
   public void resize(int width, int height) {