Fixed PWF extension test

Will merge.