diff --git a/ubuntu-kernel-cleanup.py b/ubuntu-kernel-cleanup.py
index dec4e84b50931938bd6e07c356ec783bb85bed23..486d9f62a7e2ce224bc7eeb894af6f0f4336e85e 100755
--- a/ubuntu-kernel-cleanup.py
+++ b/ubuntu-kernel-cleanup.py
@@ -48,7 +48,7 @@ def get_packages_list(pkg_match):
     installed_list = installed_out.decode('utf-8').split('\n')
     # delete empty items
     kernels = [item for item in installed_list if item != '']
-    kernel_versions = [item.replace(pkg_match, '') for item in kernels]
+    kernel_versions = [get_version(item) for item in kernels]
     # sort by version and unique
     return sorted(list(set(kernel_versions)), key=LooseVersion)