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"