Index: tests/unit_tests/test_ui_proxy.py
===================================================================
diff -u -racd09d02879e8add4cf7f694f5db8e423a76c341 -race0d2c65c6cec92c7f476df1b18043718fa2838
--- tests/unit_tests/test_ui_proxy.py	(.../test_ui_proxy.py)	(revision acd09d02879e8add4cf7f694f5db8e423a76c341)
+++ tests/unit_tests/test_ui_proxy.py	(.../test_ui_proxy.py)	(revision ace0d2c65c6cec92c7f476df1b18043718fa2838)
@@ -82,7 +82,7 @@
 
 
     def test_treatment_parameters_settings_normal(self):
-
+        test_timestamp = 0.0
         payload  = integer_to_bytearray(self.params.get("bld_flow", None))
         payload += integer_to_bytearray(self.params.get("dia_flow", None))
         payload += integer_to_bytearray(self.params.get("duration", None))
@@ -105,17 +105,17 @@
                                               message_id=MsgIds.MSG_ID_UI_NEW_TREATMENT_PARAMS.value,
                                               payload=payload)
 
-        self.assertIsNone(self.hd.ui._handler_treatment_param_settings(message))
+        self.assertIsNone(self.hd.ui._handler_treatment_param_settings(message, test_timestamp))
 
     def test_treatment_parameters_settings_wrong_type(self):
-
+        test_timestamp = 0.0
         payload = bytearray()
 
         message = DenaliMessage.build_message(channel_id=DenaliChannels.ui_to_hd_ch_id,
                                               message_id=MsgIds.MSG_ID_UI_NEW_TREATMENT_PARAMS.value,
                                               payload=payload)
 
-        self.assertRaises(struct.error, self.hd.ui._handler_treatment_param_settings, message)
+        self.assertRaises(struct.error, self.hd.ui._handler_treatment_param_settings, message, test_timestamp)
 
 
 if __name__ == '__main__':