diff --git a/20-grub.install b/20-grub.install
index ccc85fb..436360d 100755
--- a/20-grub.install
+++ b/20-grub.install
@@ -46,7 +46,7 @@ initrd /initramfs-${kernelver}.img
 options ${kernelopts}
 grub_users \$grub_users
 grub_arg --unrestricted
-grub_class kernel${flavor}
+grub_class ${ID}
 EOF
 }