Index: tst_pre_treatment_disposables/test.py
===================================================================
diff -u -rd490da73a6ff88c175db353d83f5e3f6a1e92647 -r3f9d21a04d25f785833fca1381c6f23d6364b39e
--- tst_pre_treatment_disposables/test.py	(.../test.py)	(revision d490da73a6ff88c175db353d83f5e3f6a1e92647)
+++ tst_pre_treatment_disposables/test.py	(.../test.py)	(revision 3f9d21a04d25f785833fca1381c6f23d6364b39e)
@@ -16,6 +16,7 @@
 from configuration.config import *
 from dialin.ui import utils
 from dialin.ui.hd_simulator import HDSimulator
+from dialin.common.hd_defs import PreTreatmentSubModes
  
 hd_simulator    = HDSimulator()
 
@@ -36,7 +37,7 @@
         'System Self Test' title is displayed
     """
     test.startSection("Navigating to disposables and verifying 'System Self Test' title is displayed")
-    navigate_to_pretreatment_screen(mode=3)
+    navigate_to_pretreatment_screen(PreTreatmentSubModes.HD_PRE_TREATMENT_SELF_TEST_NO_CART_STATE.value)
     test.compare(SYSTEM_SELF_TEST_TITLE, waitForObject(names.o_system_test_text).text, "'{} title  should displayed when user is navigated to disposable screen".format(SYSTEM_SELF_TEST_TITLE))
     verify_page_step_indicator(SCREEN_OBJ, PRE_TREATMENT_STEP)
     test.endSection()
@@ -190,7 +191,7 @@
         and 'NEXT' button
     """
     test.startSection("Verifying functionality of the cartridge installation screen's left, right navigation and 'NEXT' button")      
-    navigate_to_pretreatment_screen(mode=4)
+    navigate_to_pretreatment_screen(PreTreatmentSubModes.HD_PRE_TREATMENT_CART_INSTALL_STATE.value)
     verify_right_instruction_navigation(NUM_OF_CARTRIDGE_INSTALLATION_SCREENS, CARTRIDGE_INSTALLATION)
     verify_left_instruction_navigation(NUM_OF_CARTRIDGE_INSTALLATION_SCREENS, CARTRIDGE_INSTALLATION)
     verify_right_instruction_navigation(NUM_OF_CARTRIDGE_INSTALLATION_SCREENS, CARTRIDGE_INSTALLATION)
@@ -256,5 +257,4 @@
 
     utils.tstDone()
     
-    
-    
\ No newline at end of file
+        
\ No newline at end of file