diff --git a/clients/src/main/java/org/apache/kafka/common/internals/PartitionStates.java b/clients/src/main/java/org/apache/kafka/common/internals/PartitionStates.java
index 22e183ab238..daad3550738 100644
--- a/clients/src/main/java/org/apache/kafka/common/internals/PartitionStates.java
+++ b/clients/src/main/java/org/apache/kafka/common/internals/PartitionStates.java
@@ -89,7 +89,7 @@ public class PartitionStates<S> {
      * Returns the partition states in order.
      */
     public List<PartitionState<S>> partitionStates() {
-        List<PartitionState<S>> result = new ArrayList<>();
+        List<PartitionState<S>> result = new ArrayList<>(map.size());
         for (Map.Entry<TopicPartition, S> entry : map.entrySet()) {
             result.add(new PartitionState<>(entry.getKey(), entry.getValue()));
         }