fixed the stage references