diff --git a/src/pybroker/__init__.py b/src/pybroker/__init__.py
index 7e3d0aa..6bd504a 100644
--- a/src/pybroker/__init__.py
+++ b/src/pybroker/__init__.py
@@ -21,7 +21,7 @@
     enable_model_cache,
 )
 from pybroker.common import BarData, DataCol, Day, FeeMode, PriceType
-from pybroker.context import ExecContext
+from pybroker.context import ExecContext, ExecSignal, PosSizeContext
 from pybroker.config import StrategyConfig
 from pybroker.data import Alpaca, AlpacaCrypto, YFinance
 from pybroker.eval import EvalMetrics, BootstrapResult