From 68611e39d71d6f1265e2b7330b47ce41d9bda1e7 Mon Sep 17 00:00:00 2001
From: Colby Klein <shakesoda@gmail.com>
Date: Wed, 24 Sep 2008 23:17:18 +0000
Subject: [PATCH] Fix coin crash when there is a CPU player in Rave/Battle mode

---
 stepmania/src/GameState.cpp | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/stepmania/src/GameState.cpp b/stepmania/src/GameState.cpp
index 30c73c5a57..90aa8c1fe8 100644
--- a/stepmania/src/GameState.cpp
+++ b/stepmania/src/GameState.cpp
@@ -688,6 +688,17 @@ void GameState::BeginStage()
 
 void GameState::CancelStage()
 {
+	FOREACH_CpuPlayer( p )
+	{
+		switch( m_PlayMode )
+		{
+			case PLAY_MODE_BATTLE:
+			case PLAY_MODE_RAVE:
+			m_iPlayerStageTokens[p] = PREFSMAN->m_iSongsPerPlay;
+		}
+
+	}
+
 	FOREACH_EnabledPlayer( p )
 		m_iPlayerStageTokens[p] += m_iNumStagesOfThisSong;
 	m_iNumStagesOfThisSong = 0;