Update to support Mongoid 3.1.x