diff --git a/runtime/Java/src/org/antlr/v4/runtime/atn/ATN.java b/runtime/Java/src/org/antlr/v4/runtime/atn/ATN.java
index e66dc941c..279a50ef2 100644
--- a/runtime/Java/src/org/antlr/v4/runtime/atn/ATN.java
+++ b/runtime/Java/src/org/antlr/v4/runtime/atn/ATN.java
@@ -118,6 +118,13 @@ public class ATN {
 		return s.decision;
 	}
 
+    public DecisionState getDecisionState(int decision) {
+        if ( decisionToState.size()>0 ) {
+            return decisionToState.get(decision);
+        }
+        return null;
+    }
+
 	public int getNumberOfDecisions() {
 		return decisionToState.size();
 	}