From 1921bd7bdee301a51bc414793e8a2cb15c8f9f2f Mon Sep 17 00:00:00 2001
From: Jordan Halterman <jordan.halterman@gmail.com>
Date: Wed, 10 Jun 2015 20:21:21 -0700
Subject: [PATCH] Ensure all members send heartbeats to maintain sessions.

---
 .../java/net/kuujo/copycat/raft/state/RaftStateContext.java | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/raft/src/main/java/net/kuujo/copycat/raft/state/RaftStateContext.java b/raft/src/main/java/net/kuujo/copycat/raft/state/RaftStateContext.java
index f81b89b329..6037c05976 100644
--- a/raft/src/main/java/net/kuujo/copycat/raft/state/RaftStateContext.java
+++ b/raft/src/main/java/net/kuujo/copycat/raft/state/RaftStateContext.java
@@ -522,10 +522,10 @@ private CompletableFuture<Void> heartbeat(List<Member> members, CompletableFutur
    * Sends a heartbeat to a specific member.
    */
   private CompletableFuture<Void> heartbeat(Member member, List<Member> members, CompletableFuture<Void> future) {
-    KeepAliveRequest request = KeepAliveRequest.builder()
-      .withSession(getSession())
+    HeartbeatRequest request = HeartbeatRequest.builder()
+      .withMember(cluster.member().id())
       .build();
-    member.<KeepAliveRequest, KeepAliveResponse>send(request).whenComplete((response, error) -> {
+    member.<HeartbeatRequest, HeartbeatResponse>send(request).whenComplete((response, error) -> {
       threadChecker.checkThread();
       if (isOpen()) {
         if (error == null && response.status() == Response.Status.OK) {