diff --git a/tests/func/test_datachain.py b/tests/func/test_datachain.py
index 9040b778..ce7b84d9 100644
--- a/tests/func/test_datachain.py
+++ b/tests/func/test_datachain.py
@@ -24,7 +24,7 @@
 from datachain.lib.listing import LISTING_TTL, is_listing_dataset, parse_listing_uri
 from datachain.lib.tar import process_tar
 from datachain.lib.udf import Mapper
-from datachain.lib.utils import DataChainColumnError, DataChainError
+from datachain.lib.utils import DataChainError
 from datachain.query.dataset import QueryStep
 from datachain.sql.functions import path as pathfunc
 from datachain.sql.functions.array import cosine_distance, euclidean_distance