Add tests for heartbeat
Allow tests for heartbeat to pass