diff --git a/custom_components/pod_point/entity.py b/custom_components/pod_point/entity.py index cec5772..3d18d45 100644 --- a/custom_components/pod_point/entity.py +++ b/custom_components/pod_point/entity.py @@ -355,12 +355,9 @@ def __pod_image(self, model: str) -> str: img = model_type - if model_id == "03": + if model_id in ("03", "05"): img = f"{model_type}-{model_id}" - if model_id == "05": - img = "UC-05" - return f"{APP_IMAGE_URL_BASE}/{img.lower()}.png" def __model_slug(self) -> List[str]: