Fixing NullWritable issue