diff --git a/samcli/local/rapid/aws-durable-execution-emulator-arm64 b/samcli/local/rapid/aws-durable-execution-emulator-arm64
index 059c262f8b..8e17f0521a 100755
Binary files a/samcli/local/rapid/aws-durable-execution-emulator-arm64 and b/samcli/local/rapid/aws-durable-execution-emulator-arm64 differ
diff --git a/samcli/local/rapid/aws-durable-execution-emulator-x86_64 b/samcli/local/rapid/aws-durable-execution-emulator-x86_64
index 70dada5494..5f244681ac 100755
Binary files a/samcli/local/rapid/aws-durable-execution-emulator-x86_64 and b/samcli/local/rapid/aws-durable-execution-emulator-x86_64 differ
diff --git a/tests/integration/durable_integ_base.py b/tests/integration/durable_integ_base.py
index b20a435e6b..47b8cccc24 100644
--- a/tests/integration/durable_integ_base.py
+++ b/tests/integration/durable_integ_base.py
@@ -37,15 +37,6 @@ def parameterized_test_name(func, num, params):
     @classmethod
     def build_durable_functions(cls):
         """Run sam build for durable functions."""
-        # Set environment variable for SDK .whl file location
-        whl_path = Path(
-            cls.test_data_path,
-            "durable",
-            "functions",
-            "aws_durable_execution_sdk_python-1.0.0-py3-none-any.whl",
-        )
-        os.environ["DURABLE_SDK_WHL"] = str(whl_path.absolute())
-
         cls.build_dir = Path(cls.test_data_path, "durable", ".aws-sam", "build")
         cls.built_template_path = cls.build_dir / "template.yaml"
 
diff --git a/tests/integration/testdata/durable/functions/aws_durable_execution_sdk_python-1.0.0-py3-none-any.whl b/tests/integration/testdata/durable/functions/aws_durable_execution_sdk_python-1.0.0-py3-none-any.whl
deleted file mode 100644
index ba4f51513b..0000000000
Binary files a/tests/integration/testdata/durable/functions/aws_durable_execution_sdk_python-1.0.0-py3-none-any.whl and /dev/null differ
diff --git a/tests/integration/testdata/durable/functions/requirements.txt b/tests/integration/testdata/durable/functions/requirements.txt
index 45da37f681..0cf8bca12b 100644
--- a/tests/integration/testdata/durable/functions/requirements.txt
+++ b/tests/integration/testdata/durable/functions/requirements.txt
@@ -1 +1 @@
-${DURABLE_SDK_WHL}
+aws-durable-execution-sdk-python
\ No newline at end of file
diff --git a/tests/integration/testdata/durable/functions/wait_for_callback/wait_for_callback.py b/tests/integration/testdata/durable/functions/wait_for_callback/wait_for_callback.py
index 75a6bb9747..5c644c532d 100644
--- a/tests/integration/testdata/durable/functions/wait_for_callback/wait_for_callback.py
+++ b/tests/integration/testdata/durable/functions/wait_for_callback/wait_for_callback.py
@@ -2,7 +2,10 @@
 from typing import Any
 
 from aws_durable_execution_sdk_python.config import WaitForCallbackConfig
-from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.context import (
+    DurableContext,
+    WaitForCallbackContext,
+)
 from aws_durable_execution_sdk_python.execution import durable_execution
 from aws_durable_execution_sdk_python.config import Duration
 
@@ -10,7 +13,7 @@
 logger.setLevel(logging.INFO)
 
 
-def external_system_call(callback_id: str) -> None:
+def external_system_call(callback_id: str, _context: WaitForCallbackContext) -> None:
     """Simulate calling an external system with callback ID."""
     # In real usage, this would make an API call to an external system
     # passing the callback_id for the system to call back when done