Index: suite_leahi/tst_feature_options/test.py
===================================================================
diff -u -r296879f9309c86f2ac2852d3c9d76ebb4115f42c -r535c5a49b6d814e8531fa43fc816f0658d1bafef
--- suite_leahi/tst_feature_options/test.py	(.../test.py)	(revision 296879f9309c86f2ac2852d3c9d76ebb4115f42c)
+++ suite_leahi/tst_feature_options/test.py	(.../test.py)	(revision 535c5a49b6d814e8531fa43fc816f0658d1bafef)
@@ -37,7 +37,7 @@
             
     td_simulator.td_operation_mode(TDOpModes.MODE_TREA.value, 0)
     td_simulator.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value, 0, 0, 0, 0, 0, 0, 0, 0, 0
+        TDTreatmentStates.DIALYSIS_STATE.value, 0, 0, 0, 0, 0, 0, 0, 0, 0
     )
 
 
@@ -266,7 +266,7 @@
     )
 
     td_simulator.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value, 0, 0, 0, 0, 0, 0, 0, 0, 0
+        TDTreatmentStates.DIALYSIS_STATE.value, 0, 0, 0, 0, 0, 0, 0, 0, 0
     )
     headerbar_container = waitForObject(names.o_headerBar_HeaderBar, 2000)
     heparin_bar = utility.findChildByText(headerbar_container, config.HEPARIN)
@@ -319,7 +319,7 @@
     )
 
     td_simulator.td_tx_state(
-        TDTreatmentStates.TREATMENT_DIALYSIS_STATE.value, 0, 0, 0, 0, 0, 0, 0, 0, 0
+        TDTreatmentStates.DIALYSIS_STATE.value, 0, 0, 0, 0, 0, 0, 0, 0, 0
     )
 
     headerbar_container = waitForObject(names.o_headerBar_HeaderBar, 2000)