Update for mongoid 4