Fix LevelGenerator