diff --git a/end_chat.py b/end_chat.py
index 044d92c..b9554a0 100644
--- a/end_chat.py
+++ b/end_chat.py
@@ -9,7 +9,7 @@ async def main():
     # Create client connected to server at the given address
     client = await Client.connect("localhost:7233")
 
-    workflow_id = "entity-ollama-workflow"
+    workflow_id = "ollama-agent"
 
     handle = client.get_workflow_handle_for(EntityOllamaWorkflow.run, workflow_id)
 
diff --git a/get_history.py b/get_history.py
index dcdd416..0fb07c6 100644
--- a/get_history.py
+++ b/get_history.py
@@ -7,7 +7,7 @@ from workflows import EntityOllamaWorkflow
 async def main():
     # Create client connected to server at the given address
     client = await Client.connect("localhost:7233")
-    workflow_id = "entity-ollama-workflow"
+    workflow_id = "ollama-agent"
 
     handle = client.get_workflow_handle(workflow_id)
 
diff --git a/send_message.py b/send_message.py
index 3645236..d20d67d 100644
--- a/send_message.py
+++ b/send_message.py
@@ -9,7 +9,7 @@ async def main(prompt):
     # Create client connected to server at the given address
     client = await Client.connect("localhost:7233")
 
-    workflow_id = "entity-ollama-workflow"
+    workflow_id = "ollama-agent"
 
     # Sends a signal to the workflow (and starts it if needed)
     await client.start_workflow(