From a75d57f5943bdded9536b16fd7d615c299feb83d Mon Sep 17 00:00:00 2001
From: Jonas Zeunert <jonas@zeunert.org>
Date: Wed, 31 Aug 2022 15:20:03 +0200
Subject: [PATCH] puff

---
 PlayerState.py         | 5 +++--
 events/EndBallEvent.py | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/PlayerState.py b/PlayerState.py
index dd0775a..6c75a67 100644
--- a/PlayerState.py
+++ b/PlayerState.py
@@ -290,8 +290,9 @@ class PlayerState:
         if not self.isActive:
             return
 
-        self.replayBall = True
-        self.replayBallTimer = Timer(interval=1, function=self.updateReplayBall)
+        if self.ballsleft == 0 and self.bonusTime != 0:
+            self.replayBall = True
+            self.replayBallTimer = Timer(interval=1, function=self.updateReplayBall)
 
     def updateReplayBall(self):
         self.bonusTime -= 1
diff --git a/events/EndBallEvent.py b/events/EndBallEvent.py
index 03b596a..fbd550f 100644
--- a/events/EndBallEvent.py
+++ b/events/EndBallEvent.py
@@ -23,6 +23,7 @@ class EndBallEvent(Event):
         TopFlipper().deactivate()
         MainFlipper().deactivate()
 
+        self.gameState.currentPlayer.setReplayBall()
         if self.gameState.currentPlayer.replayBall or not self.gameState.currentPlayer.hasBallScored:
             OutHoleSolenoid().trigger()
             return