IEnumerable fix