Fix the scheduler unit test