diff --git a/app/src/main/java/com/fankes/miui/notify/hook/entity/SystemUIHooker.kt b/app/src/main/java/com/fankes/miui/notify/hook/entity/SystemUIHooker.kt
index 4457623..2395351 100644
--- a/app/src/main/java/com/fankes/miui/notify/hook/entity/SystemUIHooker.kt
+++ b/app/src/main/java/com/fankes/miui/notify/hook/entity/SystemUIHooker.kt
@@ -824,15 +824,10 @@ object SystemUIHooker : YukiBaseHooker() {
         }
         /** 修改 MIUI 风格通知栏的通知图标 */
         MiuiNotificationViewWrapperClass.hook {
-            /** 干掉设置通知小图标 */
+            /** 替换通知小图标 */
             injectMember {
                 method { name = "handleAppIcon" }
-                intercept()
-            }
-            /** 替换 MIUI 通知小图标 */
-            injectMember {
-                method { name = "onContentUpdated" }
-                afterHook {
+                replaceUnit {
                     field { name = "mAppIcon" }.get(instance).cast<ImageView>()?.clone {
                         compatNotifyIcon(
                             context = context,