fixed an issue where small screen edit is not showing full loading screen.