diff --git a/tck/sut_agent.py b/tck/sut_agent.py
index 936edf175..0095b4747 100644
--- a/tck/sut_agent.py
+++ b/tck/sut_agent.py
@@ -23,6 +23,7 @@
     AgentSkill,
     Message,
     Part,
+    Role,
     TaskState,
     TaskStatus,
     TaskStatusUpdateEvent,
@@ -87,7 +88,7 @@ async def execute(
             status=TaskStatus(
                 state=TaskState.TASK_STATE_WORKING,
                 message=Message(
-                    role='agent',
+                    role=Role.ROLE_AGENT,
                     message_id=str(uuid.uuid4()),
                     parts=[Part(text='Processing your question')],
                     task_id=task_id,
@@ -108,7 +109,7 @@ async def execute(
         logger.info('[SUTAgentExecutor] Response: %s', agent_reply_text)
 
         agent_message = Message(
-            role='agent',
+            role=Role.ROLE_AGENT,
             message_id=str(uuid.uuid4()),
             parts=[Part(text=agent_reply_text)],
             task_id=task_id,