Update tests import statements to use package.