Fix hang on shutdown.

1 files changed, 2 insertions, 2 deletions

diff --git a/test/test_flow.py b/test/test_flow.py
index 91e3f5f4..a3fa48d7 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -1,6 +1,6 @@
 import Queue
 from cStringIO import StringIO
-from libmproxy import console, proxy, filt, flow
+from libmproxy import console, proxy, filt, flow, controller
 import tutils
 import libpry
 
@@ -442,7 +442,7 @@ class uFlowMaster(libpry.AutoTree):
         assert fm.do_server_playback(tutils.tflow())
         q = Queue.Queue()
         fm.tick(q)
-        assert fm._shutdown
+        assert controller.exit
 
     def test_stickycookie(self):
         s = flow.State()