diff --git a/frontend/app/image/generate/page.tsx b/frontend/app/image/generate/page.tsx
index 512b07c..cc060de 100644
--- a/frontend/app/image/generate/page.tsx
+++ b/frontend/app/image/generate/page.tsx
@@ -183,12 +183,13 @@ export default function ImageGeneratePage() {
         })
       );
 
-      // When editing, append to existing images; otherwise replace
+      // When editing, replace with new edited version to maintain chain visibility
       if (editingImage) {
-        setGeneratedImages([...generatedImages, ...images]);
-        setEditingImage(null);
+        setGeneratedImages(images);
+        // Keep the editing image as the new base for next edit
+        setEditingImage(images[0]);
         setEditInstructions('');
-        toast.success('Image edited successfully!');
+        toast.success('Image edited successfully! You can continue editing.');
       } else {
         setGeneratedImages(images);
         toast.success('Images generated successfully!');
@@ -395,7 +396,7 @@ export default function ImageGeneratePage() {
         <div>
           <h2 className="text-lg font-semibold text-white mb-4">Generated Images</h2>
           {generatedImages.length > 0 ? (
-            <div className="grid grid-cols-2 gap-4">
+            <div className="space-y-4">
               {generatedImages.map((image) => (
                 <div
                   key={image.id}