diff --git a/docs/getting-started.md b/docs/getting-started.md
index 69b9c15..f89452b 100644
--- a/docs/getting-started.md
+++ b/docs/getting-started.md
@@ -168,13 +168,14 @@ pip install aws-durable-execution-sdk-python-testing
 
 ```python
 import pytest
+from aws_durable_execution_sdk_python.execution import InvocationStatus
 from my_function import handler
 
 @pytest.mark.durable_execution(handler=handler, lambda_function_name="my_function")
 def test_my_function(durable_runner):
     with durable_runner:
         result = durable_runner.run(input={"data": "test"}, timeout=10)
-    assert result.status == "SUCCEEDED"
+    assert result.status == InvocationStatus.SUCCEEDED
 ```
 
 Run tests without AWS credentials: