Fixed the model loading