Fixing for mongoid 4