Fix for the loading state check.