diff --git a/pom.xml b/pom.xml
index f1da33dd15..094aaedb22 100644
--- a/pom.xml
+++ b/pom.xml
@@ -73,7 +73,7 @@
     <findbugs-maven-plugin.version>3.0.5</findbugs-maven-plugin.version>
     <frontend-maven-plugin.version>1.8.0</frontend-maven-plugin.version>
     <guava.version>31.1-jre</guava.version>
-    <hadoop.version>3.3.4</hadoop.version>
+    <hadoop.version>3.3.6</hadoop.version>
     <jdk.tools.version>1.8</jdk.tools.version>
     <jersey.version>1.19</jersey.version>
     <jettison.version>1.5.4</jettison.version>
diff --git a/tez-dag/src/main/java/org/apache/tez/state/StateMachineTez.java b/tez-dag/src/main/java/org/apache/tez/state/StateMachineTez.java
index 3be7718926..cbb838d7f0 100644
--- a/tez-dag/src/main/java/org/apache/tez/state/StateMachineTez.java
+++ b/tez-dag/src/main/java/org/apache/tez/state/StateMachineTez.java
@@ -51,6 +51,11 @@ public STATE getCurrentState() {
     return realStatemachine.getCurrentState();
   }
 
+  @Override
+  public STATE getPreviousState() {
+    return realStatemachine.getPreviousState();
+  }
+
   @SuppressWarnings("unchecked")
   @Override
   public STATE doTransition(EVENTTYPE eventType, EVENT event) throws