Fix tests with mongoid2