Tests now expect a list back rather than a set