Fix loading screen issue.