Tiled gallery block: fix number of columns too high when removing ima…

…ges (#30582)
Automattic · Feb 6, 2019 · 8ed3aaa · 8ed3aaa
1 parent 38ac976
commit 8ed3aaa
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/client/gutenberg/extensions/tiled-gallery/edit.jsx b/client/gutenberg/extensions/tiled-gallery/edit.jsx
@@ -123,8 +123,13 @@ class TiledGalleryEdit extends Component {
 		}
 	};
 
-	onSelectImages = images =>
-		this.setAttributes( { images: images.map( image => pickRelevantMediaFiles( image ) ) } );
+	onSelectImages = images => {
+		const { columns } = this.props.attributes;
+		this.setAttributes( {
+			columns: columns ? Math.min( images.length, columns ) : columns,
+			images: images.map( image => pickRelevantMediaFiles( image ) ),
+		} );
+	};
 
 	setColumnsNumber = value => this.setAttributes( { columns: value } );