diff --git a/ast/tests/serialization/expected_leo_ast.json b/ast/tests/serialization/expected_leo_ast.json
index bdf84627c3..6227b135e6 100644
--- a/ast/tests/serialization/expected_leo_ast.json
+++ b/ast/tests/serialization/expected_leo_ast.json
@@ -11,8 +11,8 @@
       "block" : {
         "statements": [
           {
-            "Return": [
-              {
+            "Return": {
+              "expression": {
                 "Binary": {
                   "left": {
                     "Value": {
@@ -49,15 +49,21 @@
                   }
                 }
               },
-              {
+              "span" : {
                 "text": "     return 1 + 1",
                 "line": 2,
                 "start": 5,
                 "end": 17
               }
-            ]
+            }
           }
-        ]
+        ],
+        "span" : {
+          "text": "     return 1 + 1",
+          "line": 2,
+          "start": 5,
+          "end": 17
+        }
       },
       "span": {
         "text": " function main() {",