Tweak to fix error with Mongoid 3.x