Don't assume primary key is 'id'

Some models might be different