Fixing issue with level select