1 files changed, 4 insertions, 10 deletions
diff --git a/test/test_api.py b/test/test_api.py
index ecf8764..80ab62a 100755
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -1,15 +1,9 @@
 #!/usr/bin/env python3
-import sys
-import os.path
-sys.path.extend([
-    os.path.realpath(
-        os.path.join(os.path.dirname(os.path.join(__file__)), '..')
-    )
-])
+import __py_include
 
-from src.home.api import WebApiClient
-from src.home.api.types import BotType
-from src.home.config import config
+from homekit.api import WebApiClient
+from homekit.api.types import BotType
+from homekit.config import config
 
 
 if __name__ == '__main__':