Allow usage with mongoid 3.1.x