From 70063031c57cc4198e1475469610fe1d1dc2a9d9 Mon Sep 17 00:00:00 2001
From: Ludeeus <ludeeus@ludeeus.dev>
Date: Tue, 4 May 2021 20:09:00 +0000
Subject: [PATCH] spesific

---
 src/components/buttons/ha-call-service-button.js | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/components/buttons/ha-call-service-button.js b/src/components/buttons/ha-call-service-button.js
index 96666cf5d3..067ca447b0 100644
--- a/src/components/buttons/ha-call-service-button.js
+++ b/src/components/buttons/ha-call-service-button.js
@@ -70,7 +70,12 @@ class HaCallServiceButton extends EventsMixin(PolymerElement) {
           eventData.success = true;
         },
         function (err) {
-          if (err?.error?.message && err.error.message === "Connection lost") {
+          if (
+            el.domain === "homeassistant" &&
+            ["restart", "stop"].includes(el.service) &&
+            err?.error?.message &&
+            err.error.message === "Connection lost"
+          ) {
             // We expect the service call to fail with 'Connection lost' when we restart or stop
             el.$.progress.actionSuccess();
             eventData.success = true;