From 6e64ed043b4c9666b75bd2555b42968d1c2badd3 Mon Sep 17 00:00:00 2001
From: Antonio Sarosi <sarosiantonio@gmail.com>
Date: Mon, 9 Dec 2024 18:01:41 +0100
Subject: [PATCH] Fix parser test

---
 .../schema-ast/src/parser/parse_assignment.rs        | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/engine/baml-lib/schema-ast/src/parser/parse_assignment.rs b/engine/baml-lib/schema-ast/src/parser/parse_assignment.rs
index a66c64264..a0b52efc6 100644
--- a/engine/baml-lib/schema-ast/src/parser/parse_assignment.rs
+++ b/engine/baml-lib/schema-ast/src/parser/parse_assignment.rs
@@ -109,12 +109,14 @@ mod tests {
                     identifier(0, 4, [single_word(0, 4)]),
                     identifier(5, 9, [single_word(5, 9)]),
                     assignment(10, 11),
-                    field_type(12, 15, [
-                        non_union(12, 15, [
-                            identifier(12, 15, [single_word(12, 15)])
-                        ])
+                    field_type_with_attr(12, 15, [
+                        field_type(12, 15, [
+                            non_union(12, 15, [
+                                identifier(12, 15, [single_word(12, 15)])
+                            ]),
+                        ]),
                     ]),
-                ])
+                ]),
             ]
         }