diff --git a/manifests/reboot_required_notify.pp b/manifests/reboot_required_notify.pp
index 82466d2..efd8aeb 100644
--- a/manifests/reboot_required_notify.pp
+++ b/manifests/reboot_required_notify.pp
@@ -1,13 +1,6 @@
 class apt::reboot_required_notify {
 
-  $jessie_or_newer = $::operatingsystemmajrelease ? {
-    5       => false,
-    6       => false,
-    7       => false,
-    default => true,
-  }
-
-  if $jessie_or_newer {
+  if versioncmp($::operatingsystemmajrelease, 8) >= 0 {
     class { 'apt::reboot_required_notify::jessie': }
     # Clean up systems that were upgraded from Wheezy or earlier:
     class { 'apt::reboot_required_notify::wheezy': ensure => absent }