diff --git a/tests/integration_tests/modules/test_ubuntu_drivers.py b/tests/integration_tests/modules/test_ubuntu_drivers.py
index ffa19ac1282..af7e2adf5e0 100644
--- a/tests/integration_tests/modules/test_ubuntu_drivers.py
+++ b/tests/integration_tests/modules/test_ubuntu_drivers.py
@@ -13,14 +13,12 @@
         license-accepted: true
 """
 
-# NOTE(VM.GPU2.1 is not in all availability_domains: use qIZq:US-ASHBURN-AD-1)
-
 
 @pytest.mark.adhoc  # Expensive instance type
 @pytest.mark.skipif(PLATFORM != "oci", reason="Test is OCI specific")
 def test_ubuntu_drivers_installed(session_cloud: IntegrationCloud):
     with session_cloud.launch(
-        launch_kwargs={"instance_type": "VM.GPU2.1"}, user_data=USER_DATA
+        launch_kwargs={"instance_type": "VM.GPU.A10.1"}, user_data=USER_DATA
     ) as client:
         log = client.read_from_file("/var/log/cloud-init.log")
         verify_clean_log(log)
@@ -32,7 +30,7 @@ def test_ubuntu_drivers_installed(session_cloud: IntegrationCloud):
         result = client.execute("dpkg -l | grep nvidia")
         assert result.ok, "No nvidia packages found"
         assert re.search(
-            r"ii\s+linux-modules-nvidia-\d+-server", result.stdout
+            r"ii\s+nvidia.*-\d+-server", result.stdout
         ), (
             f"Did not find specific nvidia drivers packages in:"
             f" {result.stdout}"