GroupedDataFrame was not setting labels correctly