fixed imports in tests