Index: leahi_dialin/td/proxies/ui_proxy.py
===================================================================
diff -u -r272d33ed055f9d107788a17a76f2b9bc5bb510aa -r63dcccf308248c312b5f82b65f46605e6d90402d
--- leahi_dialin/td/proxies/ui_proxy.py	(.../ui_proxy.py)	(revision 272d33ed055f9d107788a17a76f2b9bc5bb510aa)
+++ leahi_dialin/td/proxies/ui_proxy.py	(.../ui_proxy.py)	(revision 63dcccf308248c312b5f82b65f46605e6d90402d)
@@ -441,15 +441,16 @@
                              can_interface = self.can_interface)
 
 
-    def cmd_send_ui_initiate_treatment(self, accepted:int = ACCEPTED ) -> None:
+    def cmd_send_ui_initiate_treatment(self, modality: int=0, accepted:int = ACCEPTED ) -> None:
         """
         Constructs and sends a UI initiate treatment message.
 
-        :param accepted: (int) the acceptance value
+        :param modality: (int) The treatment modality, HD = 0, HDF = 1, HD with HDF cartridge = 2
+        :param accepted: (int) The acceptance value
         :return: none
         """
-        acc = integer_to_bytearray(accepted)
-        payload = acc
+        payload = integer_to_bytearray(accepted)
+        payload += integer_to_bytearray(modality)
 
         cmd_generic_override(payload = payload,
                              reset = None,
@@ -474,7 +475,7 @@
         cmd_generic_override(payload = payload,
                              reset = None,
                              channel_id = CanChannels.ui_to_td_ch_id,
-                             msg_id = MsgIds.MSG_ID_UI_TREATMENT_PARAMS_CONFIRMED,
+                             msg_id = MsgIds.MSG_ID_TD_VALIDATED_TREATMENT_PARAMS,
                              entity_name = 'UI Confirm Treatment Parameters',
                              override_text = 'N/A',
                              logger = self.logger,