Index: suite_leahi/tst_main_treatment/test.py
===================================================================
diff -u -r9e9fb3fcb3825db1635bd4be27e49c5e0d354b8a -r4e2984559dbdc1dc708f2bdbc984ed03ea1fd8e1
--- suite_leahi/tst_main_treatment/test.py	(.../test.py)	(revision 9e9fb3fcb3825db1635bd4be27e49c5e0d354b8a)
+++ suite_leahi/tst_main_treatment/test.py	(.../test.py)	(revision 4e2984559dbdc1dc708f2bdbc984ed03ea1fd8e1)
@@ -338,8 +338,8 @@
         waitFor(lambda: newTreatmentDuration == 60, 5000),
         "Verifying FW received New Treatment Duration",
     )
-    td.td_duration_validate_response(0, 60, 0, 0)
-
+    
+    td.td_duration_validate_response(0, 60, 0, 0, 0)
     parent_object = waitForObject(names.o_newTreatmentDuration, 3000)
     newTreatmentDuration_value_object = utility.findObjectById(parent_object, "_value")
     test.log(f"New Treatment Duration value: {str(newTreatmentDuration_value_object.text)}")
@@ -377,9 +377,9 @@
                     0 ,
                     0)
     
-    # verify main treatment screen
+    verify main treatment screen
     test.verify(waitForObjectExists(names.mainTreatmentScreen), "In Main Treatment")
-      
+    
     # Pressure State
     test.startSection("Verifying Pressure values on main treatment screen")
     test_pressure_displayed_in_treatment_screen_range_bar()