1 files changed, 2 insertions, 1 deletions
diff --git a/demo.py b/demo.py
index bb9cb98..9626ead 100755
--- a/demo.py
+++ b/demo.py
@@ -6,7 +6,8 @@ import argparse
 from datetime import datetime
 from decimal import Decimal
 
-from finance.flow import Flow, simulate
+from finance.model import Flow
+from finance.simulate import simulate
 from finance.visualize import display