Fix level generation