fixed the import problem for tests