adapt imports in tests