Small change to the mongo queries