diff --git a/kernel/core_hook.c b/kernel/core_hook.c
index 83b75dcd..c1b44338 100644
--- a/kernel/core_hook.c
+++ b/kernel/core_hook.c
@@ -869,6 +869,9 @@ void __init ksu_core_init(void)
         pr_err("ksu_kprobe_init failed: %d\n", rc);
     }
 #endif
+    if (ksu_register_feature_handler(&kernel_umount_handler)) {
+        pr_err("Failed to register umount feature handler\n");
+    }
 }
 
 void ksu_core_exit(void)
@@ -882,4 +885,5 @@ void ksu_core_exit(void)
     pr_info("ksu_core_exit\n");
     ksu_kprobe_exit();
 #endif
-}
\ No newline at end of file
+    ksu_unregister_feature_handler(KSU_FEATURE_KERNEL_UMOUNT);
+}