diff --git a/packages/entity/src/behavior-components/expressions.jsx b/packages/entity/src/behavior-components/expressions.jsx
index 771e833..215a37e 100644
--- a/packages/entity/src/behavior-components/expressions.jsx
+++ b/packages/entity/src/behavior-components/expressions.jsx
@@ -78,35 +78,37 @@ const Expressions = ({
           );
         })
       }
-      <button
-        className="expressions__add-expression"
-        onClick={() => {
-          patch({
-            op: 'add',
-            path: join(path, 'expressions/-'),
-            value: {
-              type: 'expression',
-              ops: [
-                {
-                  type: 'key',
-                  key: 'Flow',
-                },
-                {
-                  type: 'key',
-                  key: 'nop',
-                },
-                {
-                  type: 'invoke',
-                  args: [],
-                },
-              ],
-            },
-          });
-        }}
-        type="button"
-      >
-        Add expression
-      </button>
+      <div className="expressions__expression-wrapper">
+        <button
+          className="expressions__add-expression"
+          onClick={() => {
+            patch({
+              op: 'add',
+              path: join(path, 'expressions/-'),
+              value: {
+                type: 'expression',
+                ops: [
+                  {
+                    type: 'key',
+                    key: 'Flow',
+                  },
+                  {
+                    type: 'key',
+                    key: 'nop',
+                  },
+                  {
+                    type: 'invoke',
+                    args: [],
+                  },
+                ],
+              },
+            });
+          }}
+          type="button"
+        >
+          Add expression
+        </button>
+      </div>
     </div>
   );
 };