diff --git a/dist/starline-card.js b/dist/starline-card.js
index 225ac20..e9b4697 100644
--- a/dist/starline-card.js
+++ b/dist/starline-card.js
@@ -1,6 +1,6 @@
 /**
  * lovelace-starline-card v1.2.1
- * Thu, 04 Jan 2024 12:00:03 GMT
+ * Thu, 04 Jan 2024 12:07:28 GMT
  */
 const STARLINE_ENTITIES = {
     'battery': {
@@ -744,8 +744,10 @@ class StarlineCard extends HTMLElement {
     _setInfo() {
         const $cnt = this._info.balance.element.parentNode;
         for (const name of this._config.info) {
-            const $item = $cnt.querySelector(`.info-${name}`);
-            $cnt.appendChild($item);
+            const $item = this._info[name]?.element;
+            if ($item) {
+                $cnt.appendChild($item);
+            }
         }
     }
     _initHandlers() {
diff --git a/src/ts/StarlineCard.ts b/src/ts/StarlineCard.ts
index 0de93ef..c7d4d39 100644
--- a/src/ts/StarlineCard.ts
+++ b/src/ts/StarlineCard.ts
@@ -302,8 +302,10 @@ export class StarlineCard extends HTMLElement {
         // TODO: Переделать, чтобы в _setInfoState не нужно было проверять видимость (а лучше вообще лишние элементы удалить)
         const $cnt = this._info.balance.element!.parentNode!;
         for (const name of this._config.info) {
-          const $item = $cnt.querySelector(`.info-${name}`)!;
-          $cnt.appendChild($item);
+            const $item = this._info[name]?.element;
+            if ($item) {
+                $cnt.appendChild($item);
+            }
         }
     }