diff --git a/playbooks/kayobe-overcloud-host-configure-base/tests/test_overcloud_host_configure.py b/playbooks/kayobe-overcloud-host-configure-base/tests/test_overcloud_host_configure.py
index 1a1965723..35406eefc 100644
--- a/playbooks/kayobe-overcloud-host-configure-base/tests/test_overcloud_host_configure.py
+++ b/playbooks/kayobe-overcloud-host-configure-base/tests/test_overcloud_host_configure.py
@@ -12,8 +12,8 @@
 
 
 def _is_apt():
-    info = distro.linux_distribution()
-    return info[0].startswith('Ubuntu')
+    info = distro.id()
+    return info == 'ubuntu'
 
 
 def _is_dnf():