diff --git a/system/nvidia-tkg.hook b/system/nvidia-tkg.hook
index a852812..75f556e 100644
--- a/system/nvidia-tkg.hook
+++ b/system/nvidia-tkg.hook
@@ -14,4 +14,4 @@ Target=nvidia-open-dkms-tkg
 Description=Update Nvidia modules in initcpio
 When=PostTransaction
 NeedsTargets
-Exec=/bin/sh -c 'if command -v mkinitcpio >/dev/null 2>&1; then mkinitcpio -P; else printf "\\033[31m mkinitcpio not found, please update initramfs manually\\033[0m\\n"; fi'
+Exec=/bin/sh -c 'if command -v mkinitcpio >/dev/null 2>&1; then mkinitcpio -P; elif command -v /usr/lib/booster/regenerate_images >/dev/null 2>&1; then /usr/lib/booster/regenerate_images; elif command -v dracut >/dev/null 2>&1; then dracut --regenerate-all --force; else printf "\\033[31m The initramfs generator was not found, please update initramfs manually\\033[0m\\n"; fi'