Fixed small loading issue