Fix StageScript code place