Fix broken custom collection spec that had an error due to refactoring