ImageBlock: apply width class to figure when using captions. (#1358)

WordPress · Jun 22, 2017 · 2b4f754 · 2b4f754
1 parent 5c24018
commit 2b4f754
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/blocks/library/image/index.js b/blocks/library/image/index.js
@@ -113,15 +113,15 @@ registerBlockType( 'core/image', {
 
 	save( { attributes } ) {
 		const { url, alt, caption, align = 'none' } = attributes;
-		const img = <img src={ url } alt={ alt } className={ `align${ align }` } />;
 
+		// If there's no caption set only save the image element.
 		if ( ! caption || ! caption.length ) {
-			return img;
+			return <img src={ url } alt={ alt } className={ `align${ align }` } />;
 		}
 
 		return (
-			<figure>
-				{ img }
+			<figure className={ `align${ align }` }>
+				<img src={ url } alt={ alt } />
 				<figcaption>{ caption }</figcaption>
 			</figure>
 		);