small fix to loading