diff --git a/providers/databricks/src/airflow/providers/databricks/operators/databricks.py b/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
index 826d79c4bdefd..788a12f1a3641 100644
--- a/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
+++ b/providers/databricks/src/airflow/providers/databricks/operators/databricks.py
@@ -1666,7 +1666,11 @@ class DatabricksTaskOperator(DatabricksTaskBaseOperator):
     """
 
     CALLER = "DatabricksTaskOperator"
-    template_fields = ("workflow_run_metadata",)
+    template_fields = (
+        "databricks_conn_id",
+        "task_config",
+        "workflow_run_metadata",
+    )
 
     def __init__(
         self,