Correct a bug with loading