Fix mongo tests