fixed bugs when returning to level select