Fix loading screen

Copy pasta from original source