diff --git a/llmfoundry/command_utils/data_prep/convert_delta_to_json.py b/llmfoundry/command_utils/data_prep/convert_delta_to_json.py
index d60549d1ba..2eaaa32538 100644
--- a/llmfoundry/command_utils/data_prep/convert_delta_to_json.py
+++ b/llmfoundry/command_utils/data_prep/convert_delta_to_json.py
@@ -714,11 +714,10 @@ def _check_imports():
         ) from e
 
     try:
+        from databricks import sql
         from databricks.sdk import WorkspaceClient
         from databricks.sql.client import Connection as Connection
         from databricks.sql.client import Cursor as Cursor
-
-        from databricks import sql
         _ = WorkspaceClient, Connection, Cursor, sql
     except ImportError as e:
         raise ImportError(