Fix bug for pks