diff --git a/acceptance/bundle/templates/lakeflow-integrations/output.txt b/acceptance/bundle/templates/lakeflow-integrations/output.txt
index 2a4e773c907..469c7583468 100644
--- a/acceptance/bundle/templates/lakeflow-integrations/output.txt
+++ b/acceptance/bundle/templates/lakeflow-integrations/output.txt
@@ -30,8 +30,8 @@ Validation OK!
 >>> [CLI] bundle deploy -t dev
 Building lakeflow_integrations_yaml...
 Building python_artifact...
-Uploading __generated__/my_lakeflow_integrations.examples.slack.send_slack_message.yml...
-Uploading __generated__/my_lakeflow_integrations.examples.wait_for_run_sensor.WaitForRunSensor.yml...
+Uploading __generated__/my_lakeflow_integrations.examples.slack.send_slack_message.yaml...
+Uploading __generated__/my_lakeflow_integrations.examples.wait_for_run_sensor.WaitForRunSensor.yaml...
 Uploading dist/my_lakeflow_integrations-0.0.1-py3-none-any.whl...
 Uploading bundle files to /Workspace/Users/[USERNAME]/.bundle/my_lakeflow_integrations/dev/files...
 Created jobs.send_slack_message
diff --git a/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/databricks.yml b/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/databricks.yml
index 4732a209b2b..10cbe6ca88f 100644
--- a/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/databricks.yml
+++ b/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/databricks.yml
@@ -17,7 +17,7 @@ artifacts:
     build: >
       uv run python -c "import shutil;
       shutil.rmtree('__generated__', ignore_errors=True);
-      from databricks.lakeflow.integrations.generate import main;
+      from databricks.lakeflow.integrations.generator.generate import main;
       main([
         '--package-module', 'my_lakeflow_integrations',
         '--output-dir', '__generated__',
@@ -26,7 +26,7 @@ artifacts:
         '--dependency', '${workspace.artifact_path}/.internal/my_lakeflow_integrations-0.0.1-py3-none-any.whl',
       ])"
     files:
-      - {"source": "__generated__/*.yml",}
+      - {"source": "__generated__/*.yaml",}
 
 targets:
   dev:
diff --git a/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/pyproject.toml b/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/pyproject.toml
index 5ddc44e12bf..569e73fc615 100644
--- a/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/pyproject.toml
+++ b/acceptance/bundle/templates/lakeflow-integrations/output/my_lakeflow_integrations/pyproject.toml
@@ -6,7 +6,7 @@ requires-python = ">=3.12,<3.13"
 dependencies = [
     # Any dependencies for jobs and pipelines in this project can be added here
     # See also https://docs.databricks.com/dev-tools/bundles/library-dependencies
-    "databricks-lakeflow-integrations",
+    "databricks-lakeflow-integrations==0.0.2",
 ]
 
 [tool.hatch.build.targets.wheel]
diff --git a/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/databricks.yml.tmpl b/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/databricks.yml.tmpl
index abcba39e771..2e1e8d8086f 100644
--- a/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/databricks.yml.tmpl
+++ b/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/databricks.yml.tmpl
@@ -18,7 +18,7 @@ artifacts:
     build: >
       uv run python -c "import shutil;
       shutil.rmtree('__generated__', ignore_errors=True);
-      from databricks.lakeflow.integrations.generate import main;
+      from databricks.lakeflow.integrations.generator.generate import main;
       main([
         '--package-module', '{{.project_name}}',
         '--output-dir', '__generated__',
@@ -27,7 +27,7 @@ artifacts:
         '--dependency', '${workspace.artifact_path}/.internal/{{.project_name}}-0.0.1-py3-none-any.whl',
       ])"
     files:
-      - {"source": "__generated__/*.yml",}
+      - {"source": "__generated__/*.yaml",}
 
 targets:
   dev:
diff --git a/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/pyproject.toml.tmpl b/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/pyproject.toml.tmpl
index 955d8dc2c2d..cd55e541d99 100644
--- a/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/pyproject.toml.tmpl
+++ b/libs/template/templates/lakeflow-integrations/template/{{.project_name}}/pyproject.toml.tmpl
@@ -6,7 +6,7 @@ requires-python = ">=3.12,<3.13"
 dependencies = [
     # Any dependencies for jobs and pipelines in this project can be added here
     # See also https://docs.databricks.com/dev-tools/bundles/library-dependencies
-    "databricks-lakeflow-integrations",
+    "databricks-lakeflow-integrations==0.0.2",
 ]
 
 [tool.hatch.build.targets.wheel]
diff --git a/libs/vendored_py_packages/databricks_lakeflow_integrations-0.0.1-py3-none-any.whl b/libs/vendored_py_packages/databricks_lakeflow_integrations-0.0.1-py3-none-any.whl
deleted file mode 100644
index 353677653fd..00000000000
Binary files a/libs/vendored_py_packages/databricks_lakeflow_integrations-0.0.1-py3-none-any.whl and /dev/null differ
diff --git a/libs/vendored_py_packages/databricks_lakeflow_integrations-0.0.2-py3-none-any.whl b/libs/vendored_py_packages/databricks_lakeflow_integrations-0.0.2-py3-none-any.whl
new file mode 100644
index 00000000000..6d4242a4712
Binary files /dev/null and b/libs/vendored_py_packages/databricks_lakeflow_integrations-0.0.2-py3-none-any.whl differ