diff --git a/ShortcutBadger/src/main/java/me/leolin/shortcutbadger/impl/XiaomiHomeBadger.java b/ShortcutBadger/src/main/java/me/leolin/shortcutbadger/impl/XiaomiHomeBadger.java
index c639625..b8ffa39 100644
--- a/ShortcutBadger/src/main/java/me/leolin/shortcutbadger/impl/XiaomiHomeBadger.java
+++ b/ShortcutBadger/src/main/java/me/leolin/shortcutbadger/impl/XiaomiHomeBadger.java
@@ -72,7 +72,7 @@ private void tryNewMiuiBadge(Context context, int badgeCount) throws ShortcutBad
             Notification.Builder builder = new Notification.Builder(context)
                     .setContentTitle("")
                     .setContentText("")
-                    .setSmallIcon(resolveInfo.getIconResource());
+                    .setSmallIcon(context.getApplicationInfo().icon);
             Notification notification = builder.build();
             try {
                 Field field = notification.getClass().getDeclaredField("extraNotification");