diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/ExpandExec.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/ExpandExec.scala
index d5603b3b00914..755bbdc382c5d 100644
--- a/sql/core/src/main/scala/org/apache/spark/sql/execution/ExpandExec.scala
+++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/ExpandExec.scala
@@ -177,9 +177,10 @@ case class ExpandExec(
       }
 
       s"""
-         |case $row:
-         |  ${updateCode.trim}
-         |  break;
+         |case $row: {
+         |    ${updateCode.trim}
+         |    break;
+         |  }
        """.stripMargin
     }