Index: suite_leahi/tst_main_treatment/test.py
===================================================================
diff -u -r14b4280d72f6b236cb2e2c3902e2dea5e10270be -r535c5a49b6d814e8531fa43fc816f0658d1bafef
--- suite_leahi/tst_main_treatment/test.py	(.../test.py)	(revision 14b4280d72f6b236cb2e2c3902e2dea5e10270be)
+++ suite_leahi/tst_main_treatment/test.py	(.../test.py)	(revision 535c5a49b6d814e8531fa43fc816f0658d1bafef)
@@ -367,7 +367,7 @@
     test.verify(waitForObjectExists(names.o_standByScreen_MainHome), "In Standby")
     
     td.td_operation_mode( TDOpModes.MODE_TREA.value, 0 )  
-    td.td_tx_state(TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value ,
+    td.td_tx_state(TDTreatmentStates.DIALYSIS_STATE.value ,
                     0 ,
                     0 ,
                     0 ,
@@ -419,7 +419,7 @@
     td.td_treatment_time( tx_duration_s       = 51 , 
                           tx_elapsed_s        = 5  , 
                           tx_remaining_s      = 6  )
-    td.td_tx_state(TDTreatmentStates.TREATMENT_PAUSED_STATE.value ,
+    td.td_tx_state(TDTreatmentStates.PAUSED_STATE.value ,
                     0 ,
                     0 ,
                     0 ,