diff --git a/graphql_ws/gevent.py b/graphql_ws/gevent.py index aadbe64..3089f8c 100644 --- a/graphql_ws/gevent.py +++ b/graphql_ws/gevent.py @@ -73,13 +73,14 @@ def on_start(self, connection_context, op_id, params): connection_context.request_context, params) assert isinstance(execution_result, Observable), \ "A subscription must return an observable" - execution_result.subscribe(SubscriptionObserver( + disposable = execution_result.subscribe(SubscriptionObserver( connection_context, op_id, self.send_execution_result, self.send_error, self.on_close )) + connection_context.register_operation(op_id, disposable) except Exception as e: self.send_error(connection_context, op_id, str(e))