fix bug with pk