Update for new testing package.