Fixed the model and directory