Change unit test for mongodb