Reset pk_keys in setUp
Refactor