Index: suite_leahi/tst_heparin/test.py
===================================================================
diff -u -rb6aa1adafbc0b279e012c904ecde7774fb0000ea -r535c5a49b6d814e8531fa43fc816f0658d1bafef
--- suite_leahi/tst_heparin/test.py	(.../test.py)	(revision b6aa1adafbc0b279e012c904ecde7774fb0000ea)
+++ suite_leahi/tst_heparin/test.py	(.../test.py)	(revision 535c5a49b6d814e8531fa43fc816f0658d1bafef)
@@ -42,7 +42,7 @@
 def verify_status_heparinStates(heparin_state,buttonState,status):
     test.startSection("Verify the heparin states in Heparin page")
     td.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value    ,
+        TDTreatmentStates.DIALYSIS_STATE.value    ,
         0       ,
         0       ,
         0       ,
@@ -130,8 +130,8 @@
     td.td_operation_mode(TDOpModes.MODE_PRET.value)
     td.td_blood_set_auto_load_response(vRejectionReason = 0)
     verify_create_treatment_parameters_Heparin_OFF()
-    change_treatmentstates(TDTreatmentStates.TREATMENT_PAUSED_STATE.value)
-    change_treatmentstates(TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value)
+    change_treatmentstates(TDTreatmentStates.PAUSED_STATE.value)
+    change_treatmentstates(TDTreatmentStates.DIALYSIS_STATE.value)
     waitForObject(names.o_listView_Item,1000)
     heparin = utility.setObjectText(text=config.HEPARIN, obj=names.o_listView_Item)
     mouseClick(waitForObject(heparin))
@@ -144,8 +144,8 @@
     td.td_operation_mode(TDOpModes.MODE_PRET.value)
     td.td_blood_set_auto_load_response(vRejectionReason = 0)
     utility.verify_create_treatment_parameters()
-    change_treatmentstates(TDTreatmentStates.TREATMENT_PAUSED_STATE.value)
-    change_treatmentstates(TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value)
+    change_treatmentstates(TDTreatmentStates.PAUSED_STATE.value)
+    change_treatmentstates(TDTreatmentStates.DIALYSIS_STATE.value)
     waitForObject(names.o_listView_Item,1000)
     heparin = utility.setObjectText(text=config.HEPARIN, obj=names.o_listView_Item)
     mouseClick(waitForObject(heparin))
@@ -166,7 +166,7 @@
     
     verify_status_heparinStates(2,config.RESUME,config.PAUSE)
    
-    td.td_tx_state(TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value,
+    td.td_tx_state(TDTreatmentStates.DIALYSIS_STATE.value,
                    0       ,
                    0       ,
                    0       ,