Stage init

Fixed the stage initialization problem.