Fixed loading screen in editor