Index: tests/dg_tests.py
===================================================================
diff -u -r889854fedb9ee0711460f6addd9b79e0a5ece718 -r685c0317a3c9ceb0aea35774cbc4511f3ae9fb33
--- tests/dg_tests.py	(.../dg_tests.py)	(revision 889854fedb9ee0711460f6addd9b79e0a5ece718)
+++ tests/dg_tests.py	(.../dg_tests.py)	(revision 685c0317a3c9ceb0aea35774cbc4511f3ae9fb33)
@@ -29,7 +29,7 @@
 from dialin.hd.temperatures import HDTemperaturesNames
 from dialin.dg.concentrate_pumps import DGConcentratePumpsStates
 from dialin.dg.uv_reactors import ReactorsNames
-from dialin.common.hd_defs import HDOpModes, HDOpSubModes, PreTreatmentWetSelfTestStates, PostTreatmentStates
+from dialin.common.hd_defs import HDOpModes, HDStandbyStates, PreTreatmentWetSelfTestStates, PostTreatmentStates
 from dialin.hd.post_treatment import HDPostTreatmentDrainStates
 from dialin.common.dg_defs import DGEventList
 from dialin.common.hd_defs import HDEventList
@@ -689,7 +689,7 @@
     hd.cmd_log_in_to_hd()
     sleep(1)
 
-    run_heat_disinfect()
+    #run_heat_disinfect()
 
     #run_flush_mode()
 
@@ -699,7 +699,7 @@
 
     # cmd_set_disinfect_ui_screen()
 
-    #collect_treatment_data()
+    collect_treatment_data()
 
     #collect_hd_treatment()
 
@@ -735,3 +735,5 @@
     #    print(get_ro_info())
     #    sleep(1)
 
+    #dg.heaters.cmd_set_primary_and_trimmer_heaters_target_temperature(primary_target_temp=40, trimmer_target_temp=52)
+