diff --git a/ql/src/java/org/apache/tez/dag/history/logging/proto/ProtoMessageWritable.java b/ql/src/java/org/apache/tez/dag/history/logging/proto/ProtoMessageWritable.java
index fe7cb2e0a091..0a04f6dc4de4 100644
--- a/ql/src/java/org/apache/tez/dag/history/logging/proto/ProtoMessageWritable.java
+++ b/ql/src/java/org/apache/tez/dag/history/logging/proto/ProtoMessageWritable.java
@@ -96,6 +96,9 @@ public void readFields(DataInput in) throws IOException {
       cin = CodedInputStream.newInstance(din);
       cin.setSizeLimit(Integer.MAX_VALUE);
     }
+    if (din.in != in) {
+      cin.resetSizeCounter();
+    }
     din.in = in;
     message = cin.readMessage(parser, ExtensionRegistryLite.newInstance());
   }