diff --git a/tools/test-helper/src/extension.ts b/tools/test-helper/src/extension.ts
index 6c44f75ea..0155e012d 100644
--- a/tools/test-helper/src/extension.ts
+++ b/tools/test-helper/src/extension.ts
@@ -448,7 +448,8 @@ function getWebviewContent(
         }
         .flex {
           display: flex;
-          flex-direction: column;
+          flex-wrap: wrap;
+          justify-content: center;
         }
       </style>
     </head>