From d1d85d68cb07a9bd2e82883995e6175520c85637 Mon Sep 17 00:00:00 2001
From: illuzn <57167030+illuzn@users.noreply.github.com>
Date: Sun, 19 Jun 2022 00:03:36 +0930
Subject: [PATCH] Fix logical flow

---
 apps/nspanel-lovelace-ui/luibackend/pages.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/nspanel-lovelace-ui/luibackend/pages.py b/apps/nspanel-lovelace-ui/luibackend/pages.py
index 5ccda618..5c881621 100644
--- a/apps/nspanel-lovelace-ui/luibackend/pages.py
+++ b/apps/nspanel-lovelace-ui/luibackend/pages.py
@@ -179,7 +179,10 @@ class LuiPagesGen(object):
                 else:
                     icon_color = 17299
                 if icon is not None:
-                    icon_id = get_icon_id_ha("sensor", state=status_entity.state, overwrite=icon)
+                    if status_entity is not None:
+                        icon_id = get_icon_id_ha("sensor", state=status_entity.state, overwrite=icon)
+                    else:
+                        icon_id = get_icon_id(icon)
                 return f"~button~{entityId}~{icon_id}~{icon_color}~{name}~{text}"
             else:
                 return f"~text~{entityId}~{get_icon_id('alert-circle-outline')}~17299~page not found~"