fix collection to get remove working again