diff --git a/xpdview/callbacks.py b/xpdview/callbacks.py
index 9df103c..e483a28 100644
--- a/xpdview/callbacks.py
+++ b/xpdview/callbacks.py
@@ -1,5 +1,6 @@
 from bluesky.callbacks.broker import BrokerCallbackBase
 from .waterfall import Waterfall
+import matplotlib.pyplot as plt
 
 
 class LiveWaterfall(BrokerCallbackBase):
@@ -25,11 +26,11 @@ def __init__(self, x_name, y_name, db=None, units=None):
         self.y_name = y_name
         self.units = units
 
-        from matplotlib.figure import Figure
-        self.fig = Figure()
+        self.fig = plt.figure()
 
         self.wf = Waterfall(fig=self.fig, unit=self.units)
         self.i = 0
+        self.fig.show()
 
     def start(self, doc):
         self.i = 0