diff --git a/custom_components/e3dc_rscp/diagnostics.py b/custom_components/e3dc_rscp/diagnostics.py
index 1a805dc..e83f8a4 100644
--- a/custom_components/e3dc_rscp/diagnostics.py
+++ b/custom_components/e3dc_rscp/diagnostics.py
@@ -67,6 +67,8 @@ async def async_get_config_entry_diagnostics(
 
     for bat in result["get_batteries_data"]:
         for dcb in bat["dcbs"]:
-            dcb["serialCode"] = f"{dcb['serialCode'][:3]}<redacted>"
+            bat["dcbs"][dcb][
+                "serialCode"
+            ] = f"{bat['dcbs'][dcb]['serialCode'][:3]}<redacted>"
 
     return result
diff --git a/custom_components/e3dc_rscp/manifest.json b/custom_components/e3dc_rscp/manifest.json
index 303b6a6..24cf0cc 100644
--- a/custom_components/e3dc_rscp/manifest.json
+++ b/custom_components/e3dc_rscp/manifest.json
@@ -15,6 +15,6 @@
     "pye3dc==0.7.5"
   ],
   "ssdp": [],
-  "version": "3.1.0",
+  "version": "3.1.1",
   "zeroconf": []
 }
\ No newline at end of file