diff --git a/compute/client_library/snippets/tests/test_disks.py b/compute/client_library/snippets/tests/test_disks.py
index 0627ff8e2b2..2a8fa88e79b 100644
--- a/compute/client_library/snippets/tests/test_disks.py
+++ b/compute/client_library/snippets/tests/test_disks.py
@@ -447,6 +447,7 @@ def test_create_secondary_region(
         PROJECT,
         REGION_SECONDARY,
         DISK_SIZE,
+        disk_type="pd-balanced"
     )
     assert disk.async_primary_disk.disk == autodelete_regional_blank_disk.self_link
 
@@ -460,7 +461,7 @@ def test_create_secondary(test_empty_pd_balanced_disk, autodelete_disk_name):
         secondary_disk_project=PROJECT,
         secondary_disk_zone=ZONE,
         disk_size_gb=DISK_SIZE,
-        disk_type="pd-ssd",
+        disk_type="pd-balanced",
     )
     assert disk.async_primary_disk.disk == test_empty_pd_balanced_disk.self_link
 
@@ -476,7 +477,7 @@ def test_create_custom_secondary_disk(
         secondary_disk_project=PROJECT,
         secondary_disk_zone=ZONE,
         disk_size_gb=DISK_SIZE,
-        disk_type="pd-ssd",
+        disk_type="pd-balanced",
     )
     assert disk.labels["secondary-disk-for-replication"] == "true"
     assert disk.labels["source-disk"] == test_empty_pd_balanced_disk.name
@@ -504,6 +505,7 @@ def test_start_stop_region_replication(
         PROJECT,
         REGION_SECONDARY,
         DISK_SIZE,
+        disk_type="pd-balanced"
     )
     assert start_disk_replication(
         project_id=PROJECT,
@@ -530,6 +532,7 @@ def test_start_stop_zone_replication(test_empty_pd_balanced_disk, autodelete_dis
         PROJECT,
         ZONE,
         DISK_SIZE,
+        disk_type="pd-balanced"
     )
     assert start_disk_replication(
         project_id=PROJECT,
@@ -587,6 +590,7 @@ def test_clone_disks_in_consistency_group(
         PROJECT,
         REGION_SECONDARY,
         DISK_SIZE,
+        disk_type="pd-balanced"
     )
 
     add_disk_consistency_group(
@@ -649,6 +653,7 @@ def test_stop_replications_in_consistency_group(
         PROJECT,
         REGION_SECONDARY,
         DISK_SIZE,
+        disk_type="pd-balanced"
     )
     start_disk_replication(
         project_id=PROJECT,
diff --git a/compute/client_library/snippets/tests/test_instance_from_template.py b/compute/client_library/snippets/tests/test_instance_from_template.py
index 9a3ba8775a1..f1a87da07e9 100644
--- a/compute/client_library/snippets/tests/test_instance_from_template.py
+++ b/compute/client_library/snippets/tests/test_instance_from_template.py
@@ -88,7 +88,7 @@ def test_create_instance_from_template_override(
     image_client = compute_v1.ImagesClient()
 
     image = image_client.get_from_family(
-        project="ubuntu-os-cloud", family="ubuntu-2004-lts"
+        project="ubuntu-os-cloud", family="ubuntu-2204-lts"
     )
     instance = create_instance_from_template_with_overrides(
         PROJECT,
diff --git a/compute/client_library/snippets/tests/test_instance_suspend_resume.py b/compute/client_library/snippets/tests/test_instance_suspend_resume.py
index 54ac3324a73..b5491904bd5 100644
--- a/compute/client_library/snippets/tests/test_instance_suspend_resume.py
+++ b/compute/client_library/snippets/tests/test_instance_suspend_resume.py
@@ -41,7 +41,7 @@ def _get_status(instance: compute_v1.Instance) -> compute_v1.Instance.Status:
 def compute_instance():
     instance_name = "test-instance-" + uuid.uuid4().hex[:10]
     newest_debian = get_image_from_family(
-        project="ubuntu-os-cloud", family="ubuntu-2004-lts"
+        project="ubuntu-os-cloud", family="ubuntu-2204-lts"
     )
     disk_type = f"zones/{INSTANCE_ZONE}/diskTypes/pd-standard"
     disks = [disk_from_image(disk_type, 100, True, newest_debian.self_link)]