diff --git a/src/Plugin/GraphQL/DataProducer/VertexAiSearchQuery.php b/src/Plugin/GraphQL/DataProducer/VertexAiSearchQuery.php
index f095bd28b5b57b9f733761a8d32c9f7cccf9b40f..bbd65a1a01d008b9d716e8aaac94482d28095bef 100644
--- a/src/Plugin/GraphQL/DataProducer/VertexAiSearchQuery.php
+++ b/src/Plugin/GraphQL/DataProducer/VertexAiSearchQuery.php
@@ -144,12 +144,13 @@ class VertexAiSearchQuery extends DataProducerPluginBase implements ContainerFac
     $facetParameter = $nodeQuery->getParameter('facets', 'keys');
     if (!empty($facetParameter)) {
       $facets = array_map('trim', explode(',', $facetParameter));
-      foreach ($facets as $facet) {
+      $facetSpecs = array_map(function (string $facet) {
         $key = (new FacetKey())->setKey($facet);
         $facetSpec = (new FacetSpec())->setFacetKey($key);
         $facetSpec->setLimit(50);
-        $request->setFacetSpecs([$facetSpec]);
-      }
+        return $facetSpec;
+      }, $facets);
+      $request->setFacetSpecs($facetSpecs);
     }
 
     $contentSearchSpec = new ContentSearchSpec();