Small update to get_pk in Model