Fix scheduler invocation after draining budget