Fix loading companies logic