diff --git a/custom_components/nordpool_planner/__init__.py b/custom_components/nordpool_planner/__init__.py
index c616c38..0b2c008 100644
--- a/custom_components/nordpool_planner/__init__.py
+++ b/custom_components/nordpool_planner/__init__.py
@@ -282,10 +282,11 @@ def register_output_listener_entity(self, entity, conf_key="") -> None:
     def get_device_info(self) -> DeviceInfo:
         """Get device info to group entities."""
         return DeviceInfo(
-            identifiers={(DOMAIN, self._config.data[CONF_TYPE])},
+            identifiers={(DOMAIN, self._config.entry_id)},
             name=self.name,
             manufacturer="Nordpool",
             entry_type=DeviceEntryType.SERVICE,
+            model="Forecast",
         )
 
     def input_changed(self, value):
diff --git a/custom_components/nordpool_planner/manifest.json b/custom_components/nordpool_planner/manifest.json
index 02befe3..bea76f1 100644
--- a/custom_components/nordpool_planner/manifest.json
+++ b/custom_components/nordpool_planner/manifest.json
@@ -8,5 +8,5 @@
   "iot_class": "calculated",
   "issue_tracker": "https://github.com/dala318/nordpool_planner/issues",
   "requirements": [],
-  "version": "2.0.0"
+  "version": "2.0.4"
 }
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..67296d7
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+homeassistant
\ No newline at end of file