From 786ed391daa690812baaf6b54854e622f29a4377 Mon Sep 17 00:00:00 2001
From: Alexandru Vasile <alexandru.vasile@parity.io>
Date: Tue, 16 Jan 2024 19:08:55 +0200
Subject: [PATCH] backend: Always save the waker

Signed-off-by: Alexandru Vasile <alexandru.vasile@parity.io>
---
 subxt/src/backend/unstable/follow_stream_driver.rs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/subxt/src/backend/unstable/follow_stream_driver.rs b/subxt/src/backend/unstable/follow_stream_driver.rs
index 028065f941..c97cef449b 100644
--- a/subxt/src/backend/unstable/follow_stream_driver.rs
+++ b/subxt/src/backend/unstable/follow_stream_driver.rs
@@ -221,9 +221,8 @@ impl<Hash: BlockHash> Shared<Hash> {
 
         // else, take whatever items, and save the waker if not done yet.
         let items = std::mem::take(&mut details.items);
-        if !is_done {
-            details.waker = Some(waker.clone());
-        }
+        // Always save the waker.
+        details.waker = Some(waker.clone());
         Some(items)
     }