Update entity.py

mattrayner · Sep 16, 2024 · e3ac96c · e3ac96c
1 parent e02c66d
commit e3ac96c
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/custom_components/pod_point/entity.py b/custom_components/pod_point/entity.py
@@ -355,10 +355,7 @@ def __pod_image(self, model: str) -> str:
 
         img = model_type
 
-        if model_id == "03":
-            img = f"{model_type}-{model_id}"
-
-        if model_id == "05":
+        if model_id == "03" || model_id == "05":
             img = f"{model_type}-{model_id}"
 
         return f"{APP_IMAGE_URL_BASE}/{img.lower()}.png"