diff --git a/lib/editor/tiny/plugins/aiplacement/styles.css b/lib/editor/tiny/plugins/aiplacement/styles.css
index 8e1ce78aca1..2e0a7cf4865 100644
--- a/lib/editor/tiny/plugins/aiplacement/styles.css
+++ b/lib/editor/tiny/plugins/aiplacement/styles.css
@@ -40,10 +40,6 @@
     z-index: 1000;
 }
 
-.tiny-aiplacement-image-options {
-    padding-left: 20px;
-    padding-right: 20px;
-}
 .tiny-aiplacement-generate-image,
 .tiny_aiplacement_textresponse_placeholder {
     border-radius: 15px;
@@ -54,6 +50,9 @@
     justify-content: center;
     align-items: center;
     height: 100%;
+    min-height: 150px;
+    margin-left: 0;
+    margin-top: 0;
 }
 
 .tiny_aiplacement_textresponse textarea:disabled {
@@ -211,10 +210,4 @@
         min-height: 12em;
         margin-top: 20px;
     }
-
-    .tiny-aiplacement-generate-image {
-        min-height: 200px;
-        margin-left: 15px;
-        margin-right: 15px;
-    }
 }
diff --git a/lib/editor/tiny/plugins/aiplacement/templates/modalbodyimage.mustache b/lib/editor/tiny/plugins/aiplacement/templates/modalbodyimage.mustache
index 1e7fa4adce3..57b2fc7a0d0 100644
--- a/lib/editor/tiny/plugins/aiplacement/templates/modalbodyimage.mustache
+++ b/lib/editor/tiny/plugins/aiplacement/templates/modalbodyimage.mustache
@@ -39,11 +39,11 @@
     <form class="mform" id="{{uniqid}}">
         <div id="{{elementid}}_tiny_aiplacement_overlay" class="tiny-aiplacement-overlay hidden"></div>
         <div class="row">
-            <div class="col-md-4">
+            <div class="col-xl-4 col-lg-5">
                 {{> tiny_aiplacement/imagecontrols }}
             </div>
-            <div class="col-md-8">
-                <div id="{{elementid}}_tiny_aiplacement_generate_image" class="tiny-aiplacement-generate-image text-center">
+            <div class="col-xl-8 col-lg-7">
+                <div id="{{elementid}}_tiny_aiplacement_generate_image" class="tiny-aiplacement-generate-image text-center mt-3 mt-lg-0">
                     <div class="tiny-aiplacement-fa-image"></div>
                     <div id="{{elementid}}_image_display_container" class="tiny-aiplacement-image-display-container"></div>
                 </div>