diff --git a/hub/internal/service/connection_manager/agent.go b/hub/internal/service/connection_manager/agent.go
index 74d93e5..34a008b 100644
--- a/hub/internal/service/connection_manager/agent.go
+++ b/hub/internal/service/connection_manager/agent.go
@@ -97,7 +97,7 @@ func (a *AgentConnection) listenHeartbeat(heartbeats <-chan domainHub.CreateHear
 			}
 
 			a.log.Warn().Msg("agent not send heartbeat")
-			a.cancel()
+			a.Close()
 			return
 		case heartbeat := <-heartbeats:
 			a.log.Debug().