small update to heartbeat code