Fixed minor bug from story to level selection screen