Index: tests/dg_tests.py
===================================================================
diff -u -r1c5dbffa6b205c14f92e05badc563f2a84980bfa -r0110f05bf353d1d2b38eb530c6e588e67c8f370d
--- tests/dg_tests.py	(.../dg_tests.py)	(revision 1c5dbffa6b205c14f92e05badc563f2a84980bfa)
+++ tests/dg_tests.py	(.../dg_tests.py)	(revision 0110f05bf353d1d2b38eb530c6e588e67c8f370d)
@@ -494,6 +494,8 @@
 def run_heat_disinfect():
     complete_counter = 1
     f = open("/home/fw/projects/dialin/tests/Heat_disinfect.log", "w")
+    dg.heaters.cmd_heaters_broadcast_interval_override(50)
+    sleep(1)
     dg.hd_proxy.cmd_start_stop_dg_heat_disinfect()
 
     try:
@@ -515,21 +517,21 @@
 
             print(var)
             f.write(var)
-            sleep(1)
+            sleep(0.05)
 
             # If the mode came back to standby or standby solo
-            if dg.dg_operation_mode == 3 or dg.dg_operation_mode == 4:
+            #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_dg_heat_disinfect(start=False)
+            #    if complete_counter == 1:
+            #        dg.hd_proxy.cmd_start_stop_dg_heat_disinfect(start=False)
 
                 # Write a few more complete states to make sure the complete state items are recorded
-                elif complete_counter == 3:
+            #    elif complete_counter == 3:
                     # pass
-                    f.close()
-                    break
+            #        f.close()
+            #        break
 
-                complete_counter += 1
+            #    complete_counter += 1
 
     except KeyboardInterrupt:
         dg.hd_proxy.cmd_start_stop_dg_heat_disinfect(start=False)
@@ -684,7 +686,7 @@
     hd.cmd_log_in_to_hd()
     sleep(1)
 
-    #run_heat_disinfect()
+    run_heat_disinfect()
 
     #run_flush_mode()
 
@@ -694,7 +696,7 @@
 
     # cmd_set_disinfect_ui_screen()
 
-    collect_treatment_data()
+    #collect_treatment_data()
 
     #collect_hd_treatment()