diff --git a/tests/parser.test.ts b/tests/parser.test.ts
index a8328d2..fd9c4ad 100644
--- a/tests/parser.test.ts
+++ b/tests/parser.test.ts
@@ -46,17 +46,19 @@ describe('Parser', () => {
           kind: 'FunctionDeclaration',
           name: 'add',
           parameters: ['a', 'b'],
-          body: [
-            {
-              kind: 'ReturnExpr',
-              value: {
-                kind: 'BinaryExpr',
-                operator: '+',
-                left: { kind: 'Identifier', symbol: 'a' },
-                right: { kind: 'Identifier', symbol: 'b' },
-              },
+          body: [],
+          return: {
+            kind: 'BinaryExpr',
+            operator: '+',
+            left: {
+              kind: 'Identifier',
+              symbol: 'a',
             },
-          ],
+            right: {
+              kind: 'Identifier',
+              symbol: 'b',
+            },
+          },
         },
       ],
     };