diff --git a/tests/codegen/expected_bundles/dev_bundle_monorepo.yml b/tests/codegen/expected_bundles/dev_bundle_monorepo.yml
index 91d75a77..895e557f 100644
--- a/tests/codegen/expected_bundles/dev_bundle_monorepo.yml
+++ b/tests/codegen/expected_bundles/dev_bundle_monorepo.yml
@@ -8,7 +8,11 @@ environments:
           email_notifications: null
           notification_settings: null
           webhook_notifications: null
-          health: null
+          health: 
+            rules:
+              - metric: "RUN_DURATION_SECONDS"
+                op: "GREATER_THAN" 
+                value: 7200.0
           git_source:
             git_commit: a
             git_provider: github
diff --git a/tests/codegen/expected_bundles/dev_bundle_polyrepo.yml b/tests/codegen/expected_bundles/dev_bundle_polyrepo.yml
index fda31f81..e5336a6c 100644
--- a/tests/codegen/expected_bundles/dev_bundle_polyrepo.yml
+++ b/tests/codegen/expected_bundles/dev_bundle_polyrepo.yml
@@ -8,7 +8,11 @@ environments:
           email_notifications: null
           notification_settings: null
           webhook_notifications: null
-          health: null
+          health: 
+            rules:
+              - metric: "RUN_DURATION_SECONDS"
+                op: "GREATER_THAN" 
+                value: 7200.0
           git_source:
             git_commit: a
             git_provider: github
diff --git a/tests/codegen/expected_bundles/dev_bundle_polyrepo_with_auto_libs.yml b/tests/codegen/expected_bundles/dev_bundle_polyrepo_with_auto_libs.yml
index 779e8a8f..6552fb72 100644
--- a/tests/codegen/expected_bundles/dev_bundle_polyrepo_with_auto_libs.yml
+++ b/tests/codegen/expected_bundles/dev_bundle_polyrepo_with_auto_libs.yml
@@ -65,7 +65,11 @@ environments:
           email_notifications: null
           notification_settings: null
           webhook_notifications: null
-          health: null
+          health: 
+            rules:
+              - metric: "RUN_DURATION_SECONDS"
+                op: "GREATER_THAN" 
+                value: 7200.0          
           git_source:
             git_commit: a
             git_provider: github
diff --git a/tests/codegen/expected_bundles/local_bundle.yml b/tests/codegen/expected_bundles/local_bundle.yml
index 0d6c8240..b47904c9 100644
--- a/tests/codegen/expected_bundles/local_bundle.yml
+++ b/tests/codegen/expected_bundles/local_bundle.yml
@@ -8,7 +8,11 @@ environments:
           email_notifications: null
           notification_settings: null
           webhook_notifications: null
-          health: null
+          health: 
+            rules:
+              - metric: "RUN_DURATION_SECONDS"
+                op: "GREATER_THAN" 
+                value: 7200.0
           git_source: null
           job_clusters: []
           max_concurrent_runs: 1.0
diff --git a/tests/codegen/expected_bundles/local_bundle_prefix_suffix.yml b/tests/codegen/expected_bundles/local_bundle_prefix_suffix.yml
index 97cd9daa..c5ee99f8 100644
--- a/tests/codegen/expected_bundles/local_bundle_prefix_suffix.yml
+++ b/tests/codegen/expected_bundles/local_bundle_prefix_suffix.yml
@@ -8,7 +8,11 @@ environments:
           email_notifications: null
           notification_settings: null
           webhook_notifications: null
-          health: null
+          health: 
+            rules:
+              - metric: "RUN_DURATION_SECONDS"
+                op: "GREATER_THAN" 
+                value: 7200.0
           git_source: null
           job_clusters: []
           max_concurrent_runs: 1.0
diff --git a/tests/codegen/sample_workflow.py b/tests/codegen/sample_workflow.py
index 124c4f22..1c1d1005 100644
--- a/tests/codegen/sample_workflow.py
+++ b/tests/codegen/sample_workflow.py
@@ -23,6 +23,11 @@
     run_as_user="abc@abc.com",
     tags={"test": "test2"},
     common_task_parameters={"all_tasks1": "test", "all_tasks3": "123"},  # type: ignore
+    health={
+        "rules": [
+            {"metric": "RUN_DURATION_SECONDS", "op": "GREATER_THAN", "value": 7200.0}
+        ]
+    }
 )