Fix issue with collection