diff --git a/src/datachain/__init__.py b/src/datachain/__init__.py
index 0242cdbdd..7c712f9c2 100644
--- a/src/datachain/__init__.py
+++ b/src/datachain/__init__.py
@@ -2,7 +2,7 @@
 from datachain.lib.feature import Feature
 from datachain.lib.feature_utils import pydantic_to_feature
 from datachain.lib.file import File, FileError, FileFeature, IndexedFile, TarVFile
-from datachain.lib.image import ImageFile, convert_images
+from datachain.lib.image import convert_images
 from datachain.lib.text import convert_text
 from datachain.lib.udf import Aggregator, Generator, Mapper
 from datachain.lib.utils import AbstractUDF, DataChainError
@@ -23,7 +23,6 @@
     "FileError",
     "FileFeature",
     "Generator",
-    "ImageFile",
     "IndexedFile",
     "Mapper",
     "Session",