Updated tests to refer to correct package.