diff --git a/collectors/universe_returns.py b/collectors/universe_returns.py
index 683f690..f55c6a4 100644
--- a/collectors/universe_returns.py
+++ b/collectors/universe_returns.py
@@ -212,7 +212,7 @@ def _trading_days_to_process(
     Returns ISO dates sorted chronologically. The trading-calendar module
     at the repo root handles holiday awareness (market closures through 2030).
     """
-    from trading_calendar import is_trading_day as nyse_is_trading_day
+    from alpha_engine_lib.trading_calendar import is_trading_day as nyse_is_trading_day
 
     out: list[str] = []
     d = today
diff --git a/requirements.txt b/requirements.txt
index 659b792..4e0be51 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,4 +14,4 @@ voyageai>=0.3
 jsonschema>=4.20
 arcticdb>=6.11
 # flow-doctor is pulled in transitively via alpha-engine-lib[flow_doctor].
-alpha-engine-lib[arcticdb,flow_doctor] @ git+https://github.com/cipher813/alpha-engine-lib@v0.1.1
+alpha-engine-lib[arcticdb,flow_doctor] @ git+https://github.com/cipher813/alpha-engine-lib@main