diff --git a/frontend/app_flowy/packages/flowy_sdk/lib/rust_stream.dart b/frontend/app_flowy/packages/flowy_sdk/lib/rust_stream.dart
index 633ebfe3b8..783a25ee8a 100644
--- a/frontend/app_flowy/packages/flowy_sdk/lib/rust_stream.dart
+++ b/frontend/app_flowy/packages/flowy_sdk/lib/rust_stream.dart
@@ -30,8 +30,8 @@ class RustStreamReceiver {
     return shared;
   }
 
-  static listen(void Function(SubscribeObject subject) callback) {
-    RustStreamReceiver.shared.observable.stream.listen(callback);
+  static StreamSubscription<SubscribeObject> listen(void Function(SubscribeObject subject) callback) {
+    return RustStreamReceiver.shared.observable.stream.listen(callback);
   }
 
   void streamCallback(Uint8List bytes) {