diff --git a/custom_components/nordpool_planner/__init__.py b/custom_components/nordpool_planner/__init__.py
index 1da2d79..d4a8b48 100644
--- a/custom_components/nordpool_planner/__init__.py
+++ b/custom_components/nordpool_planner/__init__.py
@@ -5,7 +5,6 @@
 import datetime as dt
 import logging
 
-from config.custom_components import nordpool
 from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
 from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN, Platform
 from homeassistant.core import HomeAssistant
@@ -248,7 +247,6 @@ def get_device_info(self) -> DeviceInfo:
             name=self.name,
             manufacturer="Nordpool",
             entry_type=DeviceEntryType.SERVICE,
-            via_device=(nordpool.DOMAIN, self._np_entity.unique_id),
         )
 
     def input_changed(self, value):
diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
index b0d2e90..e044562 100644
--- a/tests/test_config_flow.py
+++ b/tests/test_config_flow.py
@@ -2,16 +2,16 @@
 
 from unittest import mock
 
-from config.custom_components.nordpool_planner.const import *
 from custom_components.nordpool_planner import config_flow
 import pytest
-from pytest_homeassistant_custom_component.async_mock import patch
+
+# from pytest_homeassistant_custom_component.async_mock import patch
 import voluptuous as vol
 
+from custom_components.nordpool_planner.const import *
 from homeassistant import config_entries
 from homeassistant.helpers import selector
 
-
 NP_ENTITY_NAME = "sensor.nordpool_ent"
 
 SCHEMA_COPY = vol.Schema(