fix association to model and not model_id