diff --git a/frontend/app/components/banner/BannerPreview.tsx b/frontend/app/components/banner/BannerPreview.tsx
index ea333e67..4613b126 100644
--- a/frontend/app/components/banner/BannerPreview.tsx
+++ b/frontend/app/components/banner/BannerPreview.tsx
@@ -93,6 +93,7 @@ export const BannerPreview = ({
         position: 'relative',
         width: '100%',
         height: '100%',
+        flex: 1,
       }}
     />
   )
diff --git a/frontend/app/components/redesign/components/BuilderBackground.tsx b/frontend/app/components/redesign/components/BuilderBackground.tsx
index 82822a2e..71212a68 100644
--- a/frontend/app/components/redesign/components/BuilderBackground.tsx
+++ b/frontend/app/components/redesign/components/BuilderBackground.tsx
@@ -27,7 +27,6 @@ export const BuilderBackground: React.FC<BuilderBackgroundProps> = ({
   onPreviewClick,
 }) => {
   const snap = useSnapshot(toolState)
-  const isWidgetTool = snap.currentToolType === 'widget'
   const isAnimationDisabled =
     snap.currentConfig.bannerSlideAnimation === SLIDE_ANIMATION.None
 
@@ -39,11 +38,8 @@ export const BuilderBackground: React.FC<BuilderBackgroundProps> = ({
     <div
       id="builder-background"
       className={cx(
-        'bg-silver-100',
-        'rounded-[20px]',
-        'p-md',
-        'flex flex-col items-center justify-end',
-        'min-h-[600px]',
+        'bg-silver-100 rounded-[20px] p-md flex flex-col items-center min-h-[600px]',
+        onPreviewClick && 'gap-4xl',
         className,
       )}
       style={{
@@ -55,7 +51,7 @@ export const BuilderBackground: React.FC<BuilderBackgroundProps> = ({
       {onPreviewClick && !isAnimationDisabled && (
         <ToolsSecondaryButton
           icon="play"
-          className="w-[130px] order-first mb-auto"
+          className="w-[130px]"
           onClick={onPreviewClick}
         >
           Preview
@@ -64,10 +60,7 @@ export const BuilderBackground: React.FC<BuilderBackgroundProps> = ({
 
       <div
         id="browser-mockup"
-        className={cx(
-          'w-full bg-transparent rounded-2xl border border-field-border overflow-hidden flex flex-col',
-          isWidgetTool ? 'h-[752px]' : 'h-[406px]',
-        )}
+        className="w-full flex-1 rounded-2xl border border-field-border overflow-hidden flex flex-col"
       >
         <div className="flex items-center p-md bg-white">
           <div className="flex items-center gap-4 w-full">
@@ -78,7 +71,7 @@ export const BuilderBackground: React.FC<BuilderBackgroundProps> = ({
 
         <div
           id="browser-content"
-          className="flex-1 p-md bg-transparent [container-type:inline-size]"
+          className="flex-1 flex flex-col p-md [container-type:inline-size]"
         >
           {children}
         </div>
diff --git a/frontend/app/components/widget/WidgetPreview.tsx b/frontend/app/components/widget/WidgetPreview.tsx
index 5e2aac25..0c969e76 100644
--- a/frontend/app/components/widget/WidgetPreview.tsx
+++ b/frontend/app/components/widget/WidgetPreview.tsx
@@ -74,7 +74,7 @@ export const WidgetPreview = ({
       style={{
         position: 'relative',
         width: '100%',
-        height: '100%',
+        height: '678px',
         display: 'flex',
         flexDirection: 'column',
         justifyContent: 'flex-end',