Index: tests/dg_tests.py
===================================================================
diff -u -rd2a920fb57105d383a9845a4491e74b5af0f15ff -rf7388cedad143cac22f58b5cc03d26b6826c20c2
--- tests/dg_tests.py	(.../dg_tests.py)	(revision d2a920fb57105d383a9845a4491e74b5af0f15ff)
+++ tests/dg_tests.py	(.../dg_tests.py)	(revision f7388cedad143cac22f58b5cc03d26b6826c20c2)
@@ -748,7 +748,7 @@
 
     # cmd_set_disinfect_ui_screen()
 
-    collect_treatment_data()
+    #collect_treatment_data()
 
     #run_ro_permeate_sample()
 
@@ -782,3 +782,4 @@
     #    print(hd.alarms.get_alarm_state(286), hd.alarms.get_alarm_state(289), dg.alarms.get_alarm_state(286),
     #          dg.alarms.get_alarm_state(289))
     #    sleep(0.5)
+    dg.cmd_dg_set_operation_mode(1)