diff --git a/flash_examples/predict/image_embedder.py b/flash_examples/predict/image_embedder.py
index 54df44a736..db4042383d 100644
--- a/flash_examples/predict/image_embedder.py
+++ b/flash_examples/predict/image_embedder.py
@@ -33,7 +33,7 @@
 random_image = torch.randn(1, 3, 244, 244)
 
 # 6. Generate an embedding from this random image.
-embeddings = embedder.predict(random_image, data_source="tensor")
+embeddings = embedder.predict(random_image, data_source="tensors")
 
 # 7. Print embeddings shape
 print(embeddings[0].shape)