diff --git a/signalcanvas_graph_based_market_signal_s/__init__.py b/signalcanvas_graph_based_market_signal_s/__init__.py
index 3240ac9..c2902f1 100644
--- a/signalcanvas_graph_based_market_signal_s/__init__.py
+++ b/signalcanvas_graph_based_market_signal_s/__init__.py
@@ -1,5 +1,4 @@
-from .graph import Graph
-from .models import SignalNode, Link, Scenario, HedgePlan
+from .graph import Graph, SignalNode, Link, Scenario, HedgePlan
 from .registry import GraphRegistry
 from .backtester import replay_deltas
 from .privacy import aggregate_signals