puff

PlayerState.py

@@ -290,6 +290,7 @@ class PlayerState:
         if not self.isActive:
             return
 
+        if self.ballsleft == 0 and self.bonusTime != 0:
             self.replayBall = True
             self.replayBallTimer = Timer(interval=1, function=self.updateReplayBall)
 

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