Fix heartbeat to use Mongo