Index: suite_leahi/tst_solution_infusion/test.py
===================================================================
diff -u -r0e7de4782caa04625d499efc5b8f913e6f352d80 -r21c4405d2189fdbc848c24328e698034f82c977a
--- suite_leahi/tst_solution_infusion/test.py	(.../test.py)	(revision 0e7de4782caa04625d499efc5b8f913e6f352d80)
+++ suite_leahi/tst_solution_infusion/test.py	(.../test.py)	(revision 21c4405d2189fdbc848c24328e698034f82c977a)
@@ -10,6 +10,7 @@
 from leahi_dialin.utils           import conversions
 
 from leahi_dialin.common.td_defs  import TDOpModes, TDTreatmentStates
+from leahi_dialin.common.ui_defs  import TXStates
 
 # td Simulator
 from leahi_dialin.ui.td_messaging import TD_Messaging
@@ -62,7 +63,7 @@
         0       ,
         0       ,
         0       ,
-        2       , # saline in progess
+        TXStates.SALINE_BOLUS_STATE_IN_PROGRESS       , # saline in progess
         0 
     )
     
@@ -104,7 +105,7 @@
         0       ,
         0       ,
         0       ,
-        0       , # saline in idle
+        TXStates.SALINE_BOLUS_STATE_IDLE       , # saline in idle
         0 
     )