diff --git a/pyproject.toml b/pyproject.toml
index 1be4312..5c2676a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -163,12 +163,10 @@ select = [
   "F",
   "W",
   "B",
-  "B950"
 ]
 ignore = [
   "E203",
   "E501",
-  "W503",
 ]
 
 [tool.ruff.lint.flake8-pytest-style]
diff --git a/tests/common.py b/tests/common.py
index 8515338..9210da2 100644
--- a/tests/common.py
+++ b/tests/common.py
@@ -8,7 +8,7 @@
 import aiohttp
 from aioresponses import aioresponses
 
-from src.incomfortclient import HEATERLIST, Gateway, Heater
+from incomfortclient import HEATERLIST, Gateway, Heater
 
 HOSTNAME = "192.168.0.1"
 
diff --git a/tests/test_gateway.py b/tests/test_gateway.py
index 3c7f229..615bb6d 100644
--- a/tests/test_gateway.py
+++ b/tests/test_gateway.py
@@ -8,7 +8,7 @@
 import pytest
 from common import HOSTNAME, SERIAL_NO_0, SERIAL_NO_1, gwy_with_heaterlist
 
-from src.incomfortclient import (
+from incomfortclient import (
     HEATERLIST,
     NULL_SERIAL_NO,
     InvalidGateway,