diff --git a/tests/test_full_stack_contract.py b/tests/test_full_stack_contract.py
index d5705cf..8561b76 100644
--- a/tests/test_full_stack_contract.py
+++ b/tests/test_full_stack_contract.py
@@ -1,6 +1,5 @@
 import asyncio
 import json
-import os
 from pathlib import Path
 from unittest.mock import patch
 
@@ -8,6 +7,7 @@ from a2a_pack import LocalRunContext, PlatformUserAuth
 from a2a_pack.cli.local import load_local_project
 
 from agent import (
+    DecisionInput,
     InvoiceGuardStudioV1,
     _parse_browser_documents,
     _review_payload,
@@ -126,6 +126,6 @@ def test_acceptance_calls_with_mocked_persistence_reload_and_decision():
         assert reload["duplicate_count"] == 1
         assert reload["total_mismatch_count"] == 1
 
-        updated = asyncio.run(agent.record_decision(ctx, "studio-invoice-v1", {"decision_id": "d1", "decision": "hold", "note": "Investigate duplicate", "invoice_number": "INV-100"}))
+        updated = asyncio.run(agent.record_decision(ctx, "studio-invoice-v1", DecisionInput(decision_id="d1", decision="hold", note="Investigate duplicate", invoice_number="INV-100")))
         assert updated["ok"] is True
         assert updated["decisions"][0]["decision"] == "hold"