Fixing few more tests for MongoDb