Fix primary keys tests