fix api client

1 files changed, 11 insertions, 3 deletions

diff --git a/test/test_api.py b/test/test_api.py
index 959b2b3..1f6361c 100755
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -1,7 +1,15 @@
 #!/usr/bin/env python3
-from home.api import WebAPIClient
-from home.api.types import BotType
-from home.config import config
+import sys
+import os.path
+sys.path.extend([
+    os.path.realpath(
+        os.path.join(os.path.dirname(os.path.join(__file__)), '..')
+    )
+])
+
+from src.home.api import WebAPIClient
+from src.home.api.types import BotType
+from src.home.config import config
 
 
 if __name__ == '__main__':