Fix unit of measurement

dylandoamaral · Jun 4, 2024 · c9a5525 · c9a5525
1 parent 2fa088c
commit c9a5525
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/trakt_tv/sensor.py b/custom_components/trakt_tv/sensor.py
@@ -217,7 +217,11 @@ def name(self):
     @property
     def unit_of_measurement(self):
         """Return the unit of measurement of this entity"""
-        return self.title.split("_")[-1]
+        units = ["ratings", "minutes", "comments", "friends", "followers", "following", "episodes", "movies", "shows", "seasons"]
+        for unit in units:
+            if unit in self.title.lower():
+                return unit
+        return None
 
     async def async_update(self):
         """Request coordinator to update data."""