Stage.cpp fix