diff --git a/stepmania/Themes/default/Scripts/Branches.lua b/stepmania/Themes/default/Scripts/Branches.lua index b5c8dbc4db..9ca0a867fc 100644 --- a/stepmania/Themes/default/Scripts/Branches.lua +++ b/stepmania/Themes/default/Scripts/Branches.lua @@ -88,8 +88,8 @@ function GetEvaluationNextScreen( sNextScreen, sFailedScreen, sEndScreen ) return sFailedScreen end - local sType = GetTopScreenMetric( "Type" ) - if sType == "stage" then + local sIsStage = GetTopScreenMetric( "Summary" ) == "0" and not GAMESTATE:IsCourseMode() + if sIsStage then local bHasAnotherStage = GAMESTATE:HasEarnedExtraStage() bHasAnotherStage = bHasAnotherStage or not (GAMESTATE:IsFinalStage() or GAMESTATE:IsExtraStage() or GAMESTATE:IsExtraStage2() )