Index: tests/test_flush.py
===================================================================
diff -u -r3bf31442dfb7d31dac639a2c28e43055a354b236 -r7d528c2109af39ed65e9dd3d61162de89ad22d96
--- tests/test_flush.py	(.../test_flush.py)	(revision 3bf31442dfb7d31dac639a2c28e43055a354b236)
+++ tests/test_flush.py	(.../test_flush.py)	(revision 7d528c2109af39ed65e9dd3d61162de89ad22d96)
@@ -8,8 +8,9 @@
 
 def get_flush_mode_info():
 
-    info = ('State, {}, Overall_elapsed_time, {}, State_elapsed_time, {}, '
-            .format(dg.flush.flush_state, dg.flush.overall_elapsed_time, dg.flush.state_elapsed_time))
+    info = ('State, {}, Overall_elapsed_time, {}, State_elapsed_time, {}, Drain_vol, {:5.3f}, '
+            .format(dg.flush.flush_state, dg.flush.overall_elapsed_time, dg.flush.state_elapsed_time,
+                    dg.flush.flush_drain_line_volume))
     return info
 
 
@@ -81,12 +82,14 @@
                 # If it is the first call, stop heat disinfect
                 if complete_counter == 1:
                     dg.hd_proxy.cmd_start_stop_dg_flush(start=False)
-                    pass
+
                 # Write a few more complete states to make sure the complete state items are recorded
                 elif complete_counter == 3:
                     f.close()
                     break
 
+                complete_counter += 1
+
     except KeyboardInterrupt:
         dg.hd_proxy.cmd_start_stop_dg_flush(start=False)
         f.close()