[Help] Trouble with extra stage indicator

Hi friends, I'm doing a custom extra stage system via branches, when the players get the extra stage I just use AddStageToPlayer and that is working nice, the trouble is that in the Stage Indicator, It shows final stage instead of extra stage also if I use IsAnExtraStage() to check if the stage is an extra stage, it return false, there is a way to make Stepmania detect that additional stage as Extra? Or a better way to customize the conditions to obtain an extra stage?

You can either hijack the branches to prevent it from going to an Extra Stage if you don't want one or deal with the engine considering your new stage a Final Stage. The engine decides whether it is an Extra Stage by checking an internal counter that tracks whether it has awarded any Extra Stages according to its own criteria, and this counter cannot be directly read or written by Lua, so you have no way to tell the engine that it should consider the current stage an Extra Stage if it hasn't already decided that it is. You could set a flag somewhere in Lua space that indicates the current stage should be considered an Extra Stage for your theme code.