Change set to array in unittest