diff --git a/sql/server/src/main/java/org/elasticsearch/xpack/sql/querydsl/agg/PercentileRanksAgg.java b/sql/server/src/main/java/org/elasticsearch/xpack/sql/querydsl/agg/PercentileRanksAgg.java
index 5c05a9098e4..a77f87aabf3 100644
--- a/sql/server/src/main/java/org/elasticsearch/xpack/sql/querydsl/agg/PercentileRanksAgg.java
+++ b/sql/server/src/main/java/org/elasticsearch/xpack/sql/querydsl/agg/PercentileRanksAgg.java
@@ -26,9 +26,7 @@ public class PercentileRanksAgg extends LeafAgg {
 
     @Override
     AggregationBuilder toBuilder() {
-        // TODO: look at keyed
-        return percentileRanks(id())
-                .field(fieldName())
-                .values(values.stream().mapToDouble(Double::doubleValue).toArray());
+        return percentileRanks(id(), values.stream().mapToDouble(Double::doubleValue).toArray())
+                .field(fieldName());
     }
 }