fix get_pk method