Fixed primary key bug due to incorrect variable used in test unit.