Fix PlanGeneratorTest