diff --git a/docs/examples/classification.md b/docs/examples/classification.md
index 9af0a39..4d98b14 100644
--- a/docs/examples/classification.md
+++ b/docs/examples/classification.md
@@ -56,7 +56,7 @@ async function classify(data: string): Promise<SimpleClassification> {
   return classification
 }
 
-const classification = await createClassification(
+const classification = await classify(
   "Hello there I'm a nigerian prince and I want to give you money"
 )
 
@@ -99,10 +99,10 @@ async function multi_classify(data: string): Promise<MultiClassification> {
   return classification 
 }
 
-const classification = await createClassification(
+const classification = await multi_classify(
   "My account is locked and I can't access my billing info. Phone is also broken"
 )
 
 console.log({ classification })
 // { predicted_labels: [ 'billing', 'hardware' ] }
-```
\ No newline at end of file
+```