diff --git a/rust/lancedb/src/query.rs b/rust/lancedb/src/query.rs
index 23310ec5a..06ce17088 100644
--- a/rust/lancedb/src/query.rs
+++ b/rust/lancedb/src/query.rs
@@ -3229,8 +3229,9 @@ mod tests {
             ),
         ]));
         let vectors = FixedSizeListArray::from_iter_primitive::<Float32Type, _, _>(
-            rows.iter()
-                .map(|(_, _, _, vector)| Some(vector.iter().copied().map(Some).collect())),
+            rows.iter().map(|(_, _, _, vector)| {
+                Some(vector.iter().copied().map(Some).collect::<Vec<_>>())
+            }),
             2,
         );
         RecordBatch::try_new(
@@ -3304,7 +3305,7 @@ mod tests {
         let table = make_geo_query_table(&tmp_dir).await;
         let batches = table
             .query()
-            .nearest_to([0.0, 0.0])
+            .nearest_to(&[0.0, 0.0])
             .unwrap()
             .only_if(ST_DWITHIN_FILTER)
             .limit(2)
@@ -3391,11 +3392,11 @@ mod tests {
         let plain = concat_batches(&plain[0].schema(), &plain).unwrap();
         let mut plain_ids = plain["id"].as_primitive::<Int32Type>().values().to_vec();
         plain_ids.sort_unstable();
-        assert_eq!(plain_ids, vec![1, 4]);
+        assert_eq!(plain_ids, vec![1, 2, 4]);
 
         let vector = table
             .query()
-            .nearest_to([0.0, 0.0])
+            .nearest_to(&[0.0, 0.0])
             .unwrap()
             .only_if(ST_DWITHIN_FILTER)
             .limit(2)
diff --git a/rust/lancedb/src/remote/table.rs b/rust/lancedb/src/remote/table.rs
index 7c646d598..7f4d68d30 100644
--- a/rust/lancedb/src/remote/table.rs
+++ b/rust/lancedb/src/remote/table.rs
@@ -5703,7 +5703,7 @@ mod tests {
 
         let batches = table
             .query()
-            .nearest_to([0.0, 0.0])
+            .nearest_to(&[0.0, 0.0])
             .unwrap()
             .only_if(FILTER)
             .limit(2)