Update to maintain Mongoid support