Fixed the test scenarios test_NormalScheduler_CountAssignsCalculator_DifferentLabelCounts_AddNewAssigns and test_CachedScheduler_CountAssignsCalculator_DifferentLabelCounts_AddNewAssigns