diff --git a/dev/dags/example_cosmos_python_models.py b/dev/dags/example_cosmos_python_models.py
index d8083c518..cbd0ad0d2 100644
--- a/dev/dags/example_cosmos_python_models.py
+++ b/dev/dags/example_cosmos_python_models.py
@@ -21,7 +21,7 @@
 
 DEFAULT_DBT_ROOT_PATH = Path(__file__).parent / "dbt"
 DBT_ROOT_PATH = Path(os.getenv("DBT_ROOT_PATH", DEFAULT_DBT_ROOT_PATH))
-SCHEMA = "cosmos_" + os.getenv("DATABRICKS_UNIQUE_ID", "")
+SCHEMA = "_".join("cosmos", os.getenv("DATABRICKS_UNIQUE_ID", ""))
 
 # [START example_cosmos_python_models]
 example_cosmos_python_models = DbtDag(
@@ -30,7 +30,7 @@
     dbt_project_name="jaffle_shop_python",
     conn_id="databricks_default",
     profile_args={
-        "schema": "cosmos",
+        "schema": SCHEMA,
     },
     operator_args={"append_env": True},
     profile_name_override="airflow",