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 5751474..c17bb87 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
@@ -752,9 +752,9 @@ object SystemUIHooker : YukiBaseHooker() {
             /** 强制回写系统的状态栏图标样式为原生 */
             injectMember {
                 method {
-                    name = "shouldSubstituteSmallIcon"
-                    param(ExpandedNotificationClass)
-                }
+                    name { it == "shouldSubstituteSmallIcon" || it == "shouldSubstituteSmallIconForStatusBarNotification" }
+                    param { it[0] extends StatusBarNotificationClass }
+                }.all()
                 replaceToFalse()
             }
             /** 强制回写系统的状态栏图标样式为原生 */
@@ -762,7 +762,7 @@ object SystemUIHooker : YukiBaseHooker() {
                 var isUseLegacy = false
                 method {
                     name = "getSmallIcon"
-                    param(ExpandedNotificationClass, IntType)
+                    param { it[0] extends StatusBarNotificationClass && it[1] == IntType }
                 }.remedys {
                     method {
                         name = "getSmallIcon"