fix testcase to use pk