Fixed unit tests in test_mongo.py