From 17166514973842e207d2e2de7bf736f153201cf3 Mon Sep 17 00:00:00 2001
From: Andrey <andvikt@gmail.com>
Date: Thu, 18 Feb 2021 10:12:01 +0300
Subject: [PATCH] fix bugs

---
 custom_components/mega/entities.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/custom_components/mega/entities.py b/custom_components/mega/entities.py
index 87d23aa..2522256 100644
--- a/custom_components/mega/entities.py
+++ b/custom_components/mega/entities.py
@@ -50,8 +50,7 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
         super().__init__(mega.updater)
 
         self.http_cmd = http_cmd
-        if self.http_cmd == 'ds2413':
-            self.mega.ds2413_ports |= {self.port}
+
         self._state: State = None
         self.port = port
         self.config_entry = config_entry
@@ -66,6 +65,8 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
         self._customize: dict = None
         self.index = index
         self.addr = addr
+        if self.http_cmd == 'ds2413':
+            self.mega.ds2413_ports |= {self.port}
 
     @property
     def customize(self):