Index: dialin/common/hd_defs.py
===================================================================
diff -u -rbea9ebd2e6dd6ba909a7431044059a9aeb4ea0b5 -r78dd086f1f9a3ce19b8b6f7e9c2435cee3883522
--- dialin/common/hd_defs.py	(.../hd_defs.py)	(revision bea9ebd2e6dd6ba909a7431044059a9aeb4ea0b5)
+++ dialin/common/hd_defs.py	(.../hd_defs.py)	(revision 78dd086f1f9a3ce19b8b6f7e9c2435cee3883522)
@@ -79,9 +79,10 @@
     HD_PRE_TREATMENT_PRIME_STATE                            =  5    # Prime blood and dialysate circuits and run wet self-tests state
     HD_PRE_TREATMENT_RECIRCULATE_STATE                      =  6    # Re-circulate blood and dialysate circuits state
     HD_PRE_TREATMENT_PATIENT_CONNECTION_STATE               =  7    # Patient connection state
-    NUM_OF_HD_PRE_TREATMENT_STATES                          =  8    # Number of pre-treatment mode states
+    HD_PRE_TREATMENT_WET_SELF_TEST_STATE                    =  8
+    HD_PRE_TREATMENT_RSVR_STATE                             =  9
+    NUM_OF_HD_PRE_TREATMENT_STATES                          =  10   # Number of pre-treatment mode states
 
-
 @unique
 class PreTreatmentSampleWaterStates(DialinEnum):
     SAMPLE_WATER_SETUP_STATE                                =  0    # Sample water setup (flush filter) state