Index: tests/test_flush.py
===================================================================
diff -u -r01db18a18769df2048346f868412fc8a12631fc0 -r2700dbd3c0ced061a3d8141eb00e9e3a7a3e2817
--- tests/test_flush.py	(.../test_flush.py)	(revision 01db18a18769df2048346f868412fc8a12631fc0)
+++ tests/test_flush.py	(.../test_flush.py)	(revision 2700dbd3c0ced061a3d8141eb00e9e3a7a3e2817)
@@ -61,7 +61,7 @@
 
     complete_counter = 1
     f = open("/home/fw/projects/dialin/tests/Flush_mode.log", "w")
-    #dg.flush.cmd_start_stop_flush()
+    dg.hd_proxy.cmd_start_stop_dg_flush()
     try:
         while True:
             flush = get_flush_mode_info()
@@ -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.flush.cmd_start_stop_flush(start=False)
+                    dg.hd_proxy.cmd_start_stop_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.flush.cmd_start_stop_flush(start=False)
+        dg.hd_proxy.cmd_start_stop_flush(start=False)
         f.close()