Fixed loading screen bugs