stage scheduler fix