diff --git a/src/writer/wgsl/generator_impl.cc b/src/writer/wgsl/generator_impl.cc
index 4fadc64..d2afa1f 100644
--- a/src/writer/wgsl/generator_impl.cc
+++ b/src/writer/wgsl/generator_impl.cc
@@ -651,18 +651,8 @@
           if (i > 0) {
             out << ", ";
           }
-          if (auto* ident = values[i]->As<ast::IdentifierExpression>()) {
-            if (!EmitIdentifier(out, ident)) {
-              return false;
-            }
-          } else if (auto* scalar =
-                         values[i]->As<ast::ScalarConstructorExpression>()) {
-            if (!EmitScalarConstructor(out, scalar)) {
-              return false;
-            }
-          } else {
-            TINT_ICE(Writer, diagnostics_)
-                << "Unsupported workgroup_size expression";
+          if (!EmitExpression(out, values[i])) {
+            return false;
           }
         }
       }