diff --git a/optimum/utils/__init__.py b/optimum/utils/__init__.py
index db7d1f6975d..9afcdcb9117 100644
--- a/optimum/utils/__init__.py
+++ b/optimum/utils/__init__.py
@@ -33,6 +33,7 @@
     check_if_transformers_greater,
     is_accelerate_available,
     is_auto_gptq_available,
+    is_datasets_available,
     is_diffusers_available,
     is_onnx_available,
     is_onnxruntime_available,