diff --git a/src/lib/widgets/ComboWidget.svelte b/src/lib/widgets/ComboWidget.svelte
index d64163a..36a0103 100644
--- a/src/lib/widgets/ComboWidget.svelte
+++ b/src/lib/widgets/ComboWidget.svelte
@@ -20,7 +20,7 @@
  }
 </script>
 
-<div class="wrapper">
+<div class="wrapper gr-combo">
     {#if item !== null && option !== undefined}
         <label>
             <BlockTitle show_label={true}>{item.widget.name}</BlockTitle>
diff --git a/src/lib/widgets/ComfyGalleryWidget.svelte b/src/lib/widgets/ComfyGalleryWidget.svelte
index c5bacfc..ef8228a 100644
--- a/src/lib/widgets/ComfyGalleryWidget.svelte
+++ b/src/lib/widgets/ComfyGalleryWidget.svelte
@@ -33,7 +33,7 @@
  }
 
 </script>
-<div class="wrapper comfy-gallery-widget" bind:this={element}>
+<div class="wrapper comfy-gallery-widget gr-gallery" bind:this={element}>
     {#if item && itemValue}
         <Block variant="solid" padding={false}>
             <Gallery
diff --git a/src/lib/widgets/RangeWidget.svelte b/src/lib/widgets/RangeWidget.svelte
index 80c6247..805a329 100644
--- a/src/lib/widgets/RangeWidget.svelte
+++ b/src/lib/widgets/RangeWidget.svelte
@@ -22,7 +22,7 @@
  }
 </script>
 
-<div class="wrapper">
+<div class="wrapper gr-range">
     {#if item && option}
         <Range
             bind:value={option}
diff --git a/src/lib/widgets/TextWidget.svelte b/src/lib/widgets/TextWidget.svelte
index 3903588..ba3a88b 100644
--- a/src/lib/widgets/TextWidget.svelte
+++ b/src/lib/widgets/TextWidget.svelte
@@ -6,7 +6,7 @@
  $: if (item) { itemValue = item.value; }
 </script>
 
-<div class="wrapper">
+<div class="wrapper gr-textbox">
     {#if item !== null && itemValue !== null}
         <TextBox
             bind:value={$itemValue}