Fixed the package name, fixed the test