Fixed visit_model method