From b77107d60db355b78bfd3588922713c3a0c412c4 Mon Sep 17 00:00:00 2001
From: Jonas Zeunert <jonas@zeunert.org>
Date: Thu, 1 Sep 2022 01:05:17 +0200
Subject: [PATCH] go on more complexity more branches its sooo good

---
 events/EndBallEvent.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/events/EndBallEvent.py b/events/EndBallEvent.py
index 470a89c..138978e 100644
--- a/events/EndBallEvent.py
+++ b/events/EndBallEvent.py
@@ -30,13 +30,16 @@ class EndBallEvent(Event):
         TopFlipper().deactivate()
         MainFlipper().deactivate()
 
+        if self.gameState.currentPlayer.ballsLeft > 0:
+            self.gameState.currentPlayer.removeBall()
+
         self.gameState.currentPlayer.setReplayBall()
+
         if self.gameState.currentPlayer.replayBall or not self.gameState.currentPlayer.hasBallScored:
             OutHoleSolenoid().trigger()
             return
 
         self.gameState.isPlaying = False
-        self.gameState.currentPlayer.removeBall()
 
         self.scorePoints()
         self.gameState.currentPlayer.deactivate()