Index: dialin/ui/hd_simulator.py
===================================================================
diff -u -racd09d02879e8add4cf7f694f5db8e423a76c341 -r0169e7636cd2d2e1ad9e93dad8f146f52e01b0c6
--- dialin/ui/hd_simulator.py	(.../hd_simulator.py)	(revision acd09d02879e8add4cf7f694f5db8e423a76c341)
+++ dialin/ui/hd_simulator.py	(.../hd_simulator.py)	(revision 0169e7636cd2d2e1ad9e93dad8f146f52e01b0c6)
@@ -1553,7 +1553,10 @@
                                           dry_self_test_state: int,
                                           prime_state: int,
                                           recirculate_state: int,
-                                          patient_connection_state: int) -> None:
+                                          patient_connection_state: int,
+                                          wet_selftests_state: int,
+                                          pretreatment_rsrvr_state: int
+                                          ) -> None:
         """
         sends the broadcast message of the pre-treatment states
         @param sub_mode                     : (int) the main pre treatment state
@@ -1565,6 +1568,8 @@
         @param prime_state                  : (int) prime state
         @param recirculate_state            : (int) recirculate state
         @param patient_connection_state     : (int) patient connection state
+        @param wet_selftests_state          : (int) wet selftest state
+        @param pretreatment_rsrvr_state     : (int) reservoir state
         @return:
         """
         payload = integer_to_bytearray(sub_mode)
@@ -1576,6 +1581,8 @@
         payload += integer_to_bytearray(prime_state)
         payload += integer_to_bytearray(recirculate_state)
         payload += integer_to_bytearray(patient_connection_state)
+        payload += integer_to_bytearray(wet_selftests_state)
+        payload += integer_to_bytearray(pretreatment_rsrvr_state)
 
         message = DenaliMessage.build_message(channel_id=DenaliChannels.hd_sync_broadcast_ch_id,
                                               message_id=MsgIds.MSG_ID_PRE_TREATMENT_STATE.value,