Missed one change from comm_id to model_id