Fix mongoid 4