From 480c0fb91e9c891a4e6581a2ef5f2d330f6df44d Mon Sep 17 00:00:00 2001
From: vipin gupta <vipg@amazon.com>
Date: Wed, 29 Oct 2025 20:47:34 +0000
Subject: [PATCH] chore: fix step-with-retry test

---
 examples/src/step_with_retry.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/src/step_with_retry.py b/examples/src/step_with_retry.py
index bf0de0d1..43dda77f 100644
--- a/examples/src/step_with_retry.py
+++ b/examples/src/step_with_retry.py
@@ -4,6 +4,7 @@
 from aws_durable_execution_sdk_python.config import StepConfig
 from aws_durable_execution_sdk_python.context import (
     DurableContext,
+    StepContext,
     durable_step,
 )
 from aws_durable_execution_sdk_python.execution import durable_execution
@@ -14,7 +15,7 @@
 
 
 @durable_step
-def unreliable_operation() -> str:
+def unreliable_operation(_step_context: StepContext) -> str:
     failure_threshold = 0.5
     if random() > failure_threshold:  # noqa: S311
         msg = "Random error occurred"