Fix schedule_yield