CoMPaTech

diff --git a/custom_components/plugwise-beta/climate.py b/custom_components/plugwise-beta/climate.py
index ae1a1cc..41a6e6b 100644
--- a/custom_components/plugwise-beta/climate.py
+++ b/custom_components/plugwise-beta/climate.py
@@ -3,7 +3,7 @@
 import logging
 from typing import Dict

-from homeassistant.components.climate import ClimateDevice
+from homeassistant.components.climate import ClimateEntity

CoMPaTech

"""Tests for the tado integration."""

from homeassistant.components.plugwise import DOMAIN
from homeassistant.core import HomeAssistant

from tests.common import MockConfigEntry, load_fixture

from tests.test_util.aiohttp import AiohttpClientMocker

import pytest

CoMPaTech

from zeroconf import ServiceBrowser, Zeroconf


class MyListener:

    def remove_service(self, zeroconf, type, name):
        print("Service %s removed\n" % (name,))

    def add_service(self, zeroconf, type, name):
        info = zeroconf.get_service_info(type, name)

CoMPaTech

diff --git a/tests/components/plugwise/test_config_flow.py b/tests/components/plugwise/test_config_flow.py
index feb695aae8..22a6fc6079 100644
--- a/tests/components/plugwise/test_config_flow.py
+++ b/tests/components/plugwise/test_config_flow.py
@@ -2,10 +2,19 @@
 from Plugwise_Smile.Smile import Smile
 import pytest

-from homeassistant import config_entries, setup
-from homeassistant.components.plugwise.const import DOMAIN