Fix #600. Issue caused by not passing model_type argument