Fixed import of model