Removed relative import so the package will work with make test-all