Update to work with Mongoid 3.1.x.