fix imports in test