Change to 'id' instead of 'model_id' for the remaining models