diff --git a/hydra-node/src/Hydra/Node/Network.hs b/hydra-node/src/Hydra/Node/Network.hs
index 60557f37168..c9c40484549 100644
--- a/hydra-node/src/Hydra/Node/Network.hs
+++ b/hydra-node/src/Hydra/Node/Network.hs
@@ -77,9 +77,9 @@ import Hydra.Ledger (IsTx)
 import Hydra.Logging (traceWith)
 import Hydra.Logging.Messages (HydraLog (..))
 import Hydra.Network (Host (..), IP, NetworkComponent, NodeId, PortNumber)
-import Hydra.Network.Authenticate (Authenticated (Authenticated), Signed, withAuthentication)
+import Hydra.Network.Authenticate (Authenticated (..), Signed, withAuthentication)
 import Hydra.Network.Heartbeat (Heartbeat (..), withHeartbeat)
-import Hydra.Network.Message (Connectivity, Message, NetworkMessage)
+import Hydra.Network.Message (Connectivity, Message, NetworkMessage (..))
 import Hydra.Network.Ouroboros (TraceOuroborosNetwork, WithHost, withOuroborosNetwork)
 import Hydra.Network.Reliability (MessagePersistence, ReliableMsg, mkMessagePersistence, withReliability)
 import Hydra.Node (HydraNodeLog (..))
@@ -138,7 +138,9 @@ withNetwork tracer configuration callback action = do
   NetworkConfiguration{persistenceDir, signingKey, otherParties, host, port, peers, nodeId} = configuration
 
   mapHeartbeat :: Either Connectivity (Authenticated (Message tx)) -> NetworkMessage (Message tx)
-  mapHeartbeat = undefined
+  mapHeartbeat = \case
+    Left connectivity -> ConnectivityMessage connectivity
+    Right (Authenticated{payload, party}) -> OffchainProtocolMessage {sender = party, msg = payload}
 
 -- | Create `MessagePersistence` handle to be used by `Reliability` network layer.
 --