fix primary key mapping for the entity in the model

fix primary key mapping for the entity in the model #303