Bit of a fixup, missing imports.
Fixing the other pt tests to match the accumulated changes/bug-fixes