diff --git a/tests/test_finance.py b/tests/test_finance.py
index 94a98da9..5562b671 100644
--- a/tests/test_finance.py
+++ b/tests/test_finance.py
@@ -20,7 +20,6 @@ from zipline.finance.performance import PerformanceTracker
 from zipline.utils.protocol_utils import ndict
 from zipline.finance.trading import TransactionSimulator
 from zipline.utils.test_utils import \
-        drain_zipline, \
         setup_logger, \
         teardown_logger,\
         assert_single_position
diff --git a/zipline/utils/test_utils.py b/zipline/utils/test_utils.py
index 99175c63..ae804426 100644
--- a/zipline/utils/test_utils.py
+++ b/zipline/utils/test_utils.py
@@ -119,6 +119,7 @@ def drain_receiver(receiver, count=None):
 
 def assert_single_position(test, zipline, blocking=False):
     output, transaction_count = drain_zipline(test, zipline, p_blocking=blocking)
+    test.assertEqual(output[-1]['prefix'], 'DONE')
 
     test.assertEqual(
         test.zipline_test_config['order_count'],