diff --git a/src/python_testing/test_testing/TestDecorators.py b/src/python_testing/test_testing/TestDecorators.py
index 1ad5bc550bc237..82b49eddd86953 100644
--- a/src/python_testing/test_testing/TestDecorators.py
+++ b/src/python_testing/test_testing/TestDecorators.py
@@ -86,7 +86,7 @@ def test_stop(self, exception: Exception, duration: int):
     def step_skipped(self, name: str, expression: str):
         pass
 
-    def step_start(self, name: str):
+    def step_start(self, name: str, endpoint: Optional[int] = None):
         pass
 
     def step_success(self, logger, logs, duration: int, request):