[BUGFIX] Fix scheduler unit-tests