diff --git a/src/conductor/client/ai/orchestrator.py b/src/conductor/client/ai/orchestrator.py
index eff3cb34..e1a900ea 100644
--- a/src/conductor/client/ai/orchestrator.py
+++ b/src/conductor/client/ai/orchestrator.py
@@ -20,7 +20,7 @@ def __init__(self, api_configuration: Configuration, prompt_test_workflow_name:
         orkes_clients = OrkesClients(api_configuration)
 
         self.integration_client = orkes_clients.get_integration_client()
-        self.workflow_client = orkes_clients.get_integration_client()
+        self.workflow_client = orkes_clients.get_workflow_client()
         self.workflow_executor = orkes_clients.get_workflow_executor()
         self.prompt_client = orkes_clients.get_prompt_client()