diff --git a/test_pipeline.py b/test_pipeline.py
index 563c600..21ccb5c 100644
--- a/test_pipeline.py
+++ b/test_pipeline.py
@@ -9,6 +9,9 @@ import urllib.request
 import warnings
 from datetime import datetime, timezone, timedelta
 
+# Ensure src/ directory is in sys.path
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "src")))
+
 warnings.filterwarnings("ignore")
 import pgpy