fix imports in test files