diff --git a/magnum_capi_helm/driver.py b/magnum_capi_helm/driver.py
index 9fa5da5..01eecad 100644
--- a/magnum_capi_helm/driver.py
+++ b/magnum_capi_helm/driver.py
@@ -59,7 +59,7 @@ class Driver(driver.Driver):
                 "server_type": "vm",
                 # NOTE(johngarbutt) we don't depend on a specific OS,
                 # we depend on kubeadm images with cloud-init
-                "os": "capi-kubeadm-cloudinit",
+                "os": "ubuntu",
                 "coe": "kubernetes",
             },
         ]
diff --git a/magnum_capi_helm/tests/test_driver.py b/magnum_capi_helm/tests/test_driver.py
index 87a650e..2f8bf6c 100644
--- a/magnum_capi_helm/tests/test_driver.py
+++ b/magnum_capi_helm/tests/test_driver.py
@@ -51,7 +51,7 @@ class ClusterAPIDriverTest(base.DbTestCase):
             [
                 {
                     "server_type": "vm",
-                    "os": "capi-kubeadm-cloudinit",
+                    "os": "ubuntu",
                     "coe": "kubernetes",
                 }
             ],
diff --git a/magnum_capi_helm/tests/test_magnum_capi_helm.py b/magnum_capi_helm/tests/test_magnum_capi_helm.py
index 74f6010..7e2db9c 100644
--- a/magnum_capi_helm/tests/test_magnum_capi_helm.py
+++ b/magnum_capi_helm/tests/test_magnum_capi_helm.py
@@ -18,7 +18,5 @@ from magnum_capi_helm.tests import base
 
 class TestMagnumDriverLoads(base.TestCase):
     def test_get_driver(self):
-        cluster_driver = common.Driver.get_driver(
-            "vm", "capi-kubeadm-cloudinit", "kubernetes"
-        )
+        cluster_driver = common.Driver.get_driver("vm", "ubuntu", "kubernetes")
         self.assertIsInstance(cluster_driver, driver.Driver)