diff --git a/presto-main/src/main/java/io/prestosql/operator/aggregation/builder/SpillableHashAggregationBuilder.java b/presto-main/src/main/java/io/prestosql/operator/aggregation/builder/SpillableHashAggregationBuilder.java
index 3dc0a583fa1a6..44f84d6de4c7a 100644
--- a/presto-main/src/main/java/io/prestosql/operator/aggregation/builder/SpillableHashAggregationBuilder.java
+++ b/presto-main/src/main/java/io/prestosql/operator/aggregation/builder/SpillableHashAggregationBuilder.java
@@ -154,9 +154,7 @@ private boolean hasPreviousSpillCompletedSuccessfully()
     @Override
     public ListenableFuture<?> startMemoryRevoke()
     {
-        checkState(spillInProgress.isDone());
-        spillToDisk();
-        return spillInProgress;
+        return spillToDisk();
     }
 
     @Override