diff --git a/homeassistant/components/battery/conditions.yaml b/homeassistant/components/battery/conditions.yaml
index 6589e644bb3..9bd7c1f3596 100644
--- a/homeassistant/components/battery/conditions.yaml
+++ b/homeassistant/components/battery/conditions.yaml
@@ -19,6 +19,8 @@
     unit_of_measurement: "%"
   - domain: sensor
     device_class: battery
+  - domain: number
+    device_class: battery
 
 .battery_threshold_number: &battery_threshold_number
   min: 0
diff --git a/homeassistant/components/battery/triggers.yaml b/homeassistant/components/battery/triggers.yaml
index 97a5c6e5515..2ca59cf423f 100644
--- a/homeassistant/components/battery/triggers.yaml
+++ b/homeassistant/components/battery/triggers.yaml
@@ -13,6 +13,8 @@
 .battery_threshold_entity: &battery_threshold_entity
   - domain: input_number
     unit_of_measurement: "%"
+  - domain: number
+    device_class: battery
   - domain: sensor
     device_class: battery