Fixed bug with loading screen not going away