Update for MongoKit usage