Correct a bug with columnName in GroupBy