Fixing compilation issues after the removal of Partition.getKey method.