Fix another pk related issue