Fix typos with model loading process