diff --git a/grocy/schema.py b/grocy/schema.py
index 8a38efe..7896e9f 100644
--- a/grocy/schema.py
+++ b/grocy/schema.py
@@ -5,6 +5,8 @@ import logging
 SCHEMA_URL_TEMPLATE = '{domain}/api/openapi/specification'
 SCHEMA_MODEL_MAP = {
     'products': 'Product',
+    'batteries': 'Battery',
+    'battery_detail': 'BatteryDetailsResponse',
     'product_details': 'ProductDetailsResponse',
     'stock': 'StockEntry',
     'product_groups': 'ProductGroup',