diff --git a/src/KafkaFlow/Consumers/ConsumerFlowManager.cs b/src/KafkaFlow/Consumers/ConsumerFlowManager.cs
index a2e1d3a2a..770bd82a5 100644
--- a/src/KafkaFlow/Consumers/ConsumerFlowManager.cs
+++ b/src/KafkaFlow/Consumers/ConsumerFlowManager.cs
@@ -143,7 +143,6 @@ private void StartHeartbeat()
     private void StopHeartbeat()
     {
         _heartbeatTokenSource?.Cancel();
-        _heartbeatTask?.GetAwaiter().GetResult();
-        _heartbeatTask?.Dispose();
+        _heartbeatTask = null;
     }
 }