diff --git a/edb/lib/ext/ai.edgeql b/edb/lib/ext/ai.edgeql
index d4dac2bbf2db..0d14e65a9287 100644
--- a/edb/lib/ext/ai.edgeql
+++ b/edb/lib/ext/ai.edgeql
@@ -338,7 +338,7 @@ CREATE EXTENSION PACKAGE ai VERSION '1.0' {
     create function ext::ai::search(
         object: anyobject,
         query: array<std::float32>,
-    ) -> optional tuple<object: anyobject, similarity: float32>
+    ) -> optional tuple<object: anyobject, distance: float32>
     {
         create annotation std::description := '
             Search an object using its ext::ai::index index.