Update test to new Scheduler interface