Update test to remove hard-coded primary key