fix pk generation specification since we are setting it, which in turn
fixes the one failing test