diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index e35521b5..7d134fd5 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -35,4 +35,4 @@ jobs:
       - name: Run pytest
         timeout-minutes: 60
         run: |
-          pytest tests
+          PYTHONPATH=${PYTHONPATH}:custom_components/:homeassistant/tests pytest tests
diff --git a/tests/test_config_flow.py b/tests/test_config_flow.py
index 53901cf9..5666cea0 100644
--- a/tests/test_config_flow.py
+++ b/tests/test_config_flow.py
@@ -1,6 +1,6 @@
 """Test Adaptive Lighting config flow."""
 from homeassistant import data_entry_flow
-from homeassistant.components.adaptive_lighting.const import (
+from adaptive_lighting.const import (
     CONF_SUNRISE_TIME,
     CONF_SUNSET_TIME,
     DEFAULT_NAME,
diff --git a/tests/test_init.py b/tests/test_init.py
index 53f05c61..ed87a4ba 100644
--- a/tests/test_init.py
+++ b/tests/test_init.py
@@ -1,7 +1,7 @@
 """Tests for Adaptive Lighting integration."""
 from homeassistant import config_entries
-from homeassistant.components import adaptive_lighting
-from homeassistant.components.adaptive_lighting.const import (
+import adaptive_lighting
+from adaptive_lighting.const import (
     DEFAULT_NAME,
     UNDO_UPDATE_LISTENER,
 )
diff --git a/tests/test_switch.py b/tests/test_switch.py
index 68f76fda..f61170e9 100644
--- a/tests/test_switch.py
+++ b/tests/test_switch.py
@@ -6,7 +6,7 @@ from random import randint
 
 import pytest
 
-from homeassistant.components.adaptive_lighting.const import (
+from adaptive_lighting.const import (
     ADAPT_BRIGHTNESS_SWITCH,
     ADAPT_COLOR_SWITCH,
     ATTR_TURN_ON_OFF_LISTENER,
@@ -31,7 +31,7 @@ from homeassistant.components.adaptive_lighting.const import (
     SLEEP_MODE_SWITCH,
     UNDO_UPDATE_LISTENER,
 )
-from homeassistant.components.adaptive_lighting.switch import (
+from adaptive_lighting.switch import (
     _attributes_have_changed,
     _expand_light_groups,
     color_difference_redmean,