1 files changed, 2 insertions, 1 deletions
diff --git a/demo b/demo
index 91af210..1ab3a18 100755
--- a/demo
+++ b/demo
@@ -6,7 +6,8 @@ import argparse
 from datetime import datetime
 from decimal import Decimal
 
-from finance.flow import Flow, simulate, display
+from finance.flow import Flow, simulate
+from finance.visualize import display
 
 
 DESCRIPTION = "Demo application plotting financial data with finance-py"