Working for mongoid 4