diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 6ccef47..041d76c 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -28,7 +28,7 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Install Hatch 
       run: |
-        python -m pip install --upgrade hatch
+        python -m pip install hatch==1.15.0
     - uses: webfactory/ssh-agent@v0.9.1
       with:
         ssh-private-key: ${{ secrets.SDK_KEY }}
diff --git a/src/aws_durable_execution_sdk_python_testing/model.py b/src/aws_durable_execution_sdk_python_testing/model.py
index c678c8b..87f624c 100644
--- a/src/aws_durable_execution_sdk_python_testing/model.py
+++ b/src/aws_durable_execution_sdk_python_testing/model.py
@@ -2118,6 +2118,17 @@ def create_callback_event_started(cls, context: EventCreationContext) -> Event:
         callback_id: str | None = (
             callback_details.callback_id if callback_details else None
         )
+        callback_options: CallbackOptions | None = (
+            context.operation_update.callback_options
+            if context.operation_update
+            else None
+        )
+        timeout: int | None = (
+            callback_options.timeout_seconds if callback_options else None
+        )
+        heartbeat_timeout: int | None = (
+            callback_options.heartbeat_timeout_seconds if callback_options else None
+        )
         return cls(
             event_type=EventType.CALLBACK_STARTED.value,
             event_timestamp=context.start_timestamp,
@@ -2126,7 +2137,11 @@ def create_callback_event_started(cls, context: EventCreationContext) -> Event:
             operation_id=context.operation.operation_id,
             name=context.operation.name,
             parent_id=context.operation.parent_id,
-            callback_started_details=CallbackStartedDetails(callback_id=callback_id),
+            callback_started_details=CallbackStartedDetails(
+                callback_id=callback_id,
+                timeout=timeout,
+                heartbeat_timeout=heartbeat_timeout,
+            ),
         )
 
     @classmethod