diff --git a/grocy/entity.py b/grocy/entity.py
index 51bd015..714feb8 100644
--- a/grocy/entity.py
+++ b/grocy/entity.py
@@ -13,8 +13,9 @@ class Entity(object):
     SCHEMA_URL_TEMPLATE = '{domain}/api/openapi/specification'
     SCHEMA_MODEL_MAP = {
         'products': 'Product',
+        'batteries': 'Battery',
         'stock': 'StockEntry',
-        'product_groups':'ProductGroup',
+        'product_groups': 'ProductGroup',
         'locations': 'Location',
         'quantity_units': 'QuantityUnit',
         'recipes': 'Recipe'