diff --git a/homeassistant/components/button/__init__.py b/homeassistant/components/button/__init__.py
index 5dd827053a2..0e2790a2e85 100644
--- a/homeassistant/components/button/__init__.py
+++ b/homeassistant/components/button/__init__.py
@@ -92,7 +92,7 @@ class ButtonEntity(RestoreEntity):
     def _default_to_device_class_name(self) -> bool:
         """Return True if an unnamed entity should be named by its device class.
 
-        For sensors this is True if the entity has a device class.
+        For buttons this is True if the entity has a device class.
         """
         return self.device_class is not None
 
diff --git a/homeassistant/components/number/__init__.py b/homeassistant/components/number/__init__.py
index 10c296c7a7a..24c44b901a1 100644
--- a/homeassistant/components/number/__init__.py
+++ b/homeassistant/components/number/__init__.py
@@ -228,7 +228,7 @@ class NumberEntity(Entity):
     def _default_to_device_class_name(self) -> bool:
         """Return True if an unnamed entity should be named by its device class.
 
-        For sensors this is True if the entity has a device class.
+        For numbers this is True if the entity has a device class.
         """
         return self.device_class is not None