fixed loading the model