diff --git a/scripts/data_prep/convert_delta_to_json.py b/scripts/data_prep/convert_delta_to_json.py
index 5cd587ce13..277a8c1ffc 100644
--- a/scripts/data_prep/convert_delta_to_json.py
+++ b/scripts/data_prep/convert_delta_to_json.py
@@ -9,7 +9,7 @@
 from databricks.sql.client import Connection as Connection
 from databricks.sql.client import Cursor as Cursor
 
-from llmfoundry.data_prep import convert_delta_to_json_from_args
+from llmfoundry.command_utils import convert_delta_to_json_from_args
 
 MINIMUM_DB_CONNECT_DBR_VERSION = '14.1'
 MINIMUM_SQ_CONNECT_DBR_VERSION = '12.2'