fix: add reschedule on leaves, again