Fix mgo api change.