fix level gen after merging code