diff --git a/src/context.rs b/src/context.rs
index 02eda8d5c..aa6f4c49a 100644
--- a/src/context.rs
+++ b/src/context.rs
@@ -164,6 +164,7 @@ impl<A, S> HttpContext<A, S> where A: Actor<Context=Self> {
             self.stream = Some(SmallVec::new());
         }
         self.stream.as_mut().map(|s| s.push(frame));
+        self.inner.modify();
     }
 
     /// Handle of the running future
diff --git a/src/ws/context.rs b/src/ws/context.rs
index 8720b461c..56320c895 100644
--- a/src/ws/context.rs
+++ b/src/ws/context.rs
@@ -112,6 +112,7 @@ impl<A, S> WebsocketContext<A, S> where A: Actor<Context=Self> {
             }
             let stream = self.stream.as_mut().unwrap();
             stream.push(ContextFrame::Chunk(Some(data)));
+            self.inner.modify();
         } else {
             warn!("Trying to write to disconnected response");
         }
@@ -179,6 +180,7 @@ impl<A, S> WebsocketContext<A, S> where A: Actor<Context=Self> {
             self.stream = Some(SmallVec::new());
         }
         self.stream.as_mut().map(|s| s.push(frame));
+        self.inner.modify();
     }
 
     /// Handle of the running future