fix imports in test modules