Fix failing tests with mongoid 2