diff --git a/cue/format/testdata/expressions.golden b/cue/format/testdata/expressions.golden
index 5e56198..20a35a6 100644
--- a/cue/format/testdata/expressions.golden
+++ b/cue/format/testdata/expressions.golden
@@ -152,15 +152,10 @@
 		}
 	}
 
-	e: {
-		for k, v in someObject if k > "a" {
-			"\(k)": v
-		}
+	e: {for k, v in someObject if k > "a" {"\(k)": v}}
+	e: {for k, v in someObject if k > "a" {
+		"\(k)": v
 	}
-	e: {
-		for k, v in someObject if k > "a" {
-			"\(k)": v
-		}
 	}
 
 	e: {
diff --git a/cue/format/testdata/expressions.input b/cue/format/testdata/expressions.input
index e44be2c..2c65b4b 100644
--- a/cue/format/testdata/expressions.input
+++ b/cue/format/testdata/expressions.input
@@ -136,23 +136,28 @@
     for x in someObject
     if x > 9
     ]
-    "\(k)": v for k, v in someObject
-
-    "\(k)": v <-
-    for k, v in someObject
-
-    e: { "\(k)":v <-
-    for k, v in someObject
-    if k > "a"
+    for k, v in someObject {
+        "\(k)": v
+    }
+    for k, v in someObject {
+        "\(k)": v
     }
 
-    e: { "\(k)":v for k, v in someObject if k > "a"}
-    e: { "\(k)":v <-
-    for k, v in someObject if k > "a"}
+    e: {
+        for k, v in someObject
+        if k > "a" {
+            "\(k)":v
+        }
+    }
+
+    e: { for k, v in someObject if k > "a" {"\(k)":v} }
+    e: { for k, v in someObject if k > "a" {
+        "\(k)":v }}
     
-    e: { "\(k)":v <-
+    e: {
     for k, v in someObject
-    if k > "a"}
+    if k > "a" {
+        "\(k)":v}}
 
     e: [{
         a: 1, b: 2,