small loading fix to make it work