Update for mongo driver 1.5.4