diff --git a/src/common/entity/domain_icon.ts b/src/common/entity/domain_icon.ts
index f950dd3dbe..af56f1135b 100644
--- a/src/common/entity/domain_icon.ts
+++ b/src/common/entity/domain_icon.ts
@@ -108,9 +108,9 @@ export const domainIcon = (
     case "switch":
       switch (stateObj?.attributes.device_class) {
         case "outlet":
-          return state === "on" ? mdiPowerPlug : mdiPowerPlugOff;
+          return compareState === "on" ? mdiPowerPlug : mdiPowerPlugOff;
         case "switch":
-          return state === "on" ? mdiToggleSwitch : mdiToggleSwitchOff;
+          return compareState === "on" ? mdiToggleSwitch : mdiToggleSwitchOff;
         default:
           return mdiFlash;
       }