Switch tests to be in the punkt package