From 060843860fe300f8448d0d2476de8963d5ddf5a2 Mon Sep 17 00:00:00 2001
From: Franck Nijhof <git@frenck.dev>
Date: Fri, 21 May 2021 13:18:45 +0200
Subject: [PATCH] Fix Coronavirus

---
 homeassistant/components/coronavirus/sensor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/homeassistant/components/coronavirus/sensor.py b/homeassistant/components/coronavirus/sensor.py
index 92d75d4b8d4..ec28f5a1d9b 100644
--- a/homeassistant/components/coronavirus/sensor.py
+++ b/homeassistant/components/coronavirus/sensor.py
@@ -38,6 +38,8 @@ class CoronavirusSensor(CoordinatorEntity, SensorEntity):
         else:
             self.name = f"{coordinator.data[country].country} Coronavirus {info_type}"
         self.unique_id = f"{country}-{info_type}"
+        self.country = country
+        self.info_type = info_type
 
         self._icon = SENSORS[self.info_type]
         self._unit_of_measurement = "people"