diff --git a/SA2InstantWinButton/SA2InstantWinButton.cpp b/SA2InstantWinButton/SA2InstantWinButton.cpp
index 1d72f00..813729a 100644
--- a/SA2InstantWinButton/SA2InstantWinButton.cpp
+++ b/SA2InstantWinButton/SA2InstantWinButton.cpp
@@ -8,7 +8,7 @@ extern "C"
 {
 	__declspec(dllexport) void OnFrame()
 	{
-		if (GameMode == GameMode_Level && (ControllerPointers[0]->press & Buttons_X))
+		if (GameState == GameStates_Ingame && (ControllerPointers[0]->press & Buttons_X))
 		{
 			if (ControllerPointers[0]->press & Buttons_B)
 				AwardWin(1);