Fix test failing in mongo