diff --git a/tree-sitter-p4/grammar.js b/tree-sitter-p4/grammar.js
index a7370ce..a921c8a 100644
--- a/tree-sitter-p4/grammar.js
+++ b/tree-sitter-p4/grammar.js
@@ -94,7 +94,7 @@ export default grammar({
         transitionSelectionExpression: $ => choice($.identifier, $.selectExpression),
         keysetExpression: $ => $.expression,
         binaryOperatorExpression: $ => choice($.binaryEqualOperatorExpression),
-        arrayAccessExpression: $=> seq($.identifier, $.open_bracket, $.expression, $.close_bracket),
+        arrayAccessExpression: $=> seq($.expression, $.open_bracket, $.expression, $.close_bracket),
 
         // Binary Operations
         binaryEqualOperatorExpression: $ => prec.left(2, seq($.expression, $.double_equal, $.expression)),
diff --git a/tree-sitter-p4/test/corpus/expressions.txt b/tree-sitter-p4/test/corpus/expressions.txt
index 61a710b..29993f1 100644
--- a/tree-sitter-p4/test/corpus/expressions.txt
+++ b/tree-sitter-p4/test/corpus/expressions.txt
@@ -144,7 +144,9 @@ parser simple() {
        (expressionStatement
         (expression
          (arrayAccessExpression
-          (identifier)
+          (expression
+           (identifier)
+          )
           (open_bracket)
           (expression
            (integer)