fix groupByKey