diff --git a/examples/workers/ai/demo.py b/examples/workers/ai/demo.py
index 00f68eeaf2f..ba29c427ee6 100644
--- a/examples/workers/ai/demo.py
+++ b/examples/workers/ai/demo.py
@@ -1,19 +1,21 @@
 import os
 import sys
 from cloudflare import Cloudflare
-from cloudflare.types.workers.ai_run_params import Translation
+from cloudflare.types.workers.ai_run_response import Translation
 
 account_id = os.getenv("CLOUDFLARE_ACCOUNT_ID")
 if account_id is None:
     sys.exit("CLOUDFLARE_ACCOUNT_ID is not defined")
 
+
 client = Cloudflare()
 
-input=Translation(
+t = client.workers.ai.run(
+    "@cf/meta/m2m100-1.2b",
+    account_id=account_id,
     text="I'll have an order of the moule frites",
-    source_lang="english",
     target_lang="french",
+    source_lang="english",
 )
 
-t = client.workers.ai.run("@cf/meta/m2m100-1.2b", account_id=account_id, body=input)
-print(t)
+print(t['translated_text'])