Minor fix for heartbeat test