diff --git a/core/trino-main/src/main/java/io/trino/operator/aggregation/SumDataSizeForStats.java b/core/trino-main/src/main/java/io/trino/operator/aggregation/SumDataSizeForStats.java
index 60f8b8d0542e..44a95f947bc2 100644
--- a/core/trino-main/src/main/java/io/trino/operator/aggregation/SumDataSizeForStats.java
+++ b/core/trino-main/src/main/java/io/trino/operator/aggregation/SumDataSizeForStats.java
@@ -37,7 +37,7 @@ private SumDataSizeForStats() {}
 
     @InputFunction
     @TypeParameter("T")
-    public static void input(@AggregationState LongState state, @BlockPosition @SqlType("T") Block block, @BlockIndex int index)
+    public static void input(@AggregationState LongState state, @NullablePosition @BlockPosition @SqlType("T") Block block, @BlockIndex int index)
     {
         update(state, block.getEstimatedDataSizeForStats(index));
     }