diff --git a/ui/src/pages/AgentDetail.tsx b/ui/src/pages/AgentDetail.tsx
index 90cf2d99..ab0e6338 100644
--- a/ui/src/pages/AgentDetail.tsx
+++ b/ui/src/pages/AgentDetail.tsx
@@ -289,6 +289,7 @@ export function AgentDetail() {
       queryClient.invalidateQueries({ queryKey: queryKeys.agents.taskSessions(agentId!) });
       if (selectedCompanyId) {
         queryClient.invalidateQueries({ queryKey: queryKeys.agents.list(selectedCompanyId) });
+        queryClient.invalidateQueries({ queryKey: queryKeys.heartbeats(selectedCompanyId, agentId!) });
       }
       if (action === "invoke" && data && typeof data === "object" && "id" in data) {
         navigate(`/agents/${agentId}/runs/${(data as HeartbeatRun).id}`);