Use correct collection class