diff --git a/packages/squiggle-lang/__tests__/ast/parse_test.ts b/packages/squiggle-lang/__tests__/ast/parse_test.ts
index e2c4588506..f84cb30ac8 100644
--- a/packages/squiggle-lang/__tests__/ast/parse_test.ts
+++ b/packages/squiggle-lang/__tests__/ast/parse_test.ts
@@ -168,11 +168,11 @@ describe("Peggy parse", () => {
     );
     testParse(
       "x :: 1 / kg = 1",
-      "(Program (LetStatement :x (UnitTypeSignature (InfixUnitType / 1 :kg)) 1))"
+      "(Program (LetStatement :x (UnitTypeSignature (InfixUnitType / 1 :kg)) 1 (Decorator :unitType '1 / kg')))"
     );
     testParse(
       "x :: 1 = 2",
-      "(Program (LetStatement :x (UnitTypeSignature 1) 2))"
+      "(Program (LetStatement :x (UnitTypeSignature 1) 2 (Decorator :unitType '1')))"
     );
     testParse(
       "x :: kg*m/s = 1",