diff --git a/parser/python.lalrpop b/parser/python.lalrpop
index 90f4ce2..604a4f9 100644
--- a/parser/python.lalrpop
+++ b/parser/python.lalrpop
@@ -372,7 +372,7 @@ IfStatement: ast::Stmt = {
             let x = ast::Stmt {
                 custom: (),
                 location: i.0,
-                end_location: i.4.last().unwrap().end_location,
+                end_location,
                 node: ast::StmtKind::If { test: Box::new(i.2), body: i.4, orelse: last },
             };
             last = vec![x];
diff --git a/parser/src/snapshots/rustpython_parser__parser__tests__parse_if_elif_else.snap b/parser/src/snapshots/rustpython_parser__parser__tests__parse_if_elif_else.snap
index 7f493dd..26bd492 100644
--- a/parser/src/snapshots/rustpython_parser__parser__tests__parse_if_elif_else.snap
+++ b/parser/src/snapshots/rustpython_parser__parser__tests__parse_if_elif_else.snap
@@ -79,8 +79,8 @@ expression: parse_ast
                     },
                     end_location: Some(
                         Location {
-                            row: 2,
-                            column: 10,
+                            row: 3,
+                            column: 8,
                         },
                     ),
                     custom: (),