diff --git a/sdk-http-vertx/src/main/java/dev/restate/sdk/http/vertx/HttpResponseFlowAdapter.java b/sdk-http-vertx/src/main/java/dev/restate/sdk/http/vertx/HttpResponseFlowAdapter.java
index 95f82b5c..d69bd0ec 100644
--- a/sdk-http-vertx/src/main/java/dev/restate/sdk/http/vertx/HttpResponseFlowAdapter.java
+++ b/sdk-http-vertx/src/main/java/dev/restate/sdk/http/vertx/HttpResponseFlowAdapter.java
@@ -40,7 +40,7 @@ public void onSubscribe(Flow.Subscription subscription) {
 
   @Override
   public void onNext(ByteBuffer byteBuffer) {
-    if (this.httpServerResponse.closed()) {
+    if (this.httpServerResponse.ended()) {
       cancelSubscription();
       return;
     }
@@ -87,8 +87,8 @@ private void endResponse() {
   }
 
   private void cancelSubscription() {
-    LOG.trace("Cancelling subscription");
     if (this.outputSubscription != null) {
+      LOG.trace("Cancelling subscription");
       Flow.Subscription outputSubscription = this.outputSubscription;
       this.outputSubscription = null;
       outputSubscription.cancel();