Fixed loading screen bug