diff --git a/src/microsoft-extensions-ai/ollama/OllamaExamples/TextEmbeddingCaching.cs b/src/microsoft-extensions-ai/ollama/OllamaExamples/TextEmbeddingCaching.cs
index 25f9fa1..7a12525 100644
--- a/src/microsoft-extensions-ai/ollama/OllamaExamples/TextEmbeddingCaching.cs
+++ b/src/microsoft-extensions-ai/ollama/OllamaExamples/TextEmbeddingCaching.cs
@@ -23,7 +23,7 @@ public static async Task TextEmbeddingCaching()
 
         foreach(var prompt in prompts)
         {
-            var embedding = await generator.GenerateEmbeddingVectorAsync("What is AI?");
+            var embedding = await generator.GenerateEmbeddingVectorAsync(prompt);
 
             Console.WriteLine(string.Join(", ", embedding.ToArray()));
         }