diff --git a/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py b/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py
index 9de652bc3..1fe9d3ff9 100644
--- a/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py
+++ b/examples/databricks_agent/databricks_agent/databricks_agent_example_usage.py
@@ -11,7 +11,7 @@
 
 import flytekit
 from flytekit import Resources, task, workflow
-from flytekitplugins.spark import Databricks
+from flytekitplugins.spark import DatabricksV2 as Databricks
 
 
 # %% [markdown]