Index: suite_leahi/tst_fluid_bolus/test.py
===================================================================
diff -u -r8db28688e2a5c59581798955c6127a63addf100b -r535c5a49b6d814e8531fa43fc816f0658d1bafef
--- suite_leahi/tst_fluid_bolus/test.py	(.../test.py)	(revision 8db28688e2a5c59581798955c6127a63addf100b)
+++ suite_leahi/tst_fluid_bolus/test.py	(.../test.py)	(revision 535c5a49b6d814e8531fa43fc816f0658d1bafef)
@@ -44,7 +44,7 @@
     test.verify(waitForObjectExists(names.o_standByScreen_MainHome), "In Standby")
     td_interface.td_operation_mode( TDOpModes.MODE_TREA.value, 0 )
     td_interface.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value    ,
+        TDTreatmentStates.DIALYSIS_STATE.value    ,
         0       ,
         0       ,
         0       ,
@@ -81,7 +81,7 @@
 
     # change state to in progress
     td_interface.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value    ,
+        TDTreatmentStates.DIALYSIS_STATE.value    ,
         0       ,
         0       ,
         0       ,
@@ -123,7 +123,7 @@
 
     # change state to idle
     td_interface.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value    ,
+        TDTreatmentStates.DIALYSIS_STATE.value    ,
         0       ,
         0       ,
         0       ,