diff --git a/packages/core/package.json b/packages/core/package.json
index 5ffbd91d8..0a6664512 100644
--- a/packages/core/package.json
+++ b/packages/core/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@viamrobotics/prime-core",
-  "version": "0.0.170",
+  "version": "0.0.171",
   "repository": {
     "type": "git",
     "url": "https://github.com/viamrobotics/prime.git",
diff --git a/packages/core/src/lib/toggle-button.svelte b/packages/core/src/lib/toggle-button.svelte
index ec33a329a..b4a8f2e7e 100644
--- a/packages/core/src/lib/toggle-button.svelte
+++ b/packages/core/src/lib/toggle-button.svelte
@@ -57,7 +57,7 @@ const handleClick = (value: string) => {
 
 <div
   aria-disabled={disabled ? true : undefined}
-  class={cx('flex', extraClasses)}
+  class={cx('flex-col', extraClasses)}
   {...$$restProps}
 >
   {#if $$slots.legend}