Index: tests/test_flush.py
===================================================================
diff -u -r2700dbd3c0ced061a3d8141eb00e9e3a7a3e2817 -r3bf31442dfb7d31dac639a2c28e43055a354b236
--- tests/test_flush.py	(.../test_flush.py)	(revision 2700dbd3c0ced061a3d8141eb00e9e3a7a3e2817)
+++ tests/test_flush.py	(.../test_flush.py)	(revision 3bf31442dfb7d31dac639a2c28e43055a354b236)
@@ -80,15 +80,15 @@
             if dg.dg_operation_mode == 3 or dg.dg_operation_mode == 4:
                 # If it is the first call, stop heat disinfect
                 if complete_counter == 1:
-                    dg.hd_proxy.cmd_start_stop_flush(start=False)
+                    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
 
     except KeyboardInterrupt:
-        dg.hd_proxy.cmd_start_stop_flush(start=False)
+        dg.hd_proxy.cmd_start_stop_dg_flush(start=False)
         f.close()