Address remounting concern

WordPress · Jan 19, 2024 · b20a114 · b20a114
1 parent 0273411
commit b20a114
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/packages/edit-site/src/hooks/template-part-edit.js b/packages/edit-site/src/hooks/template-part-edit.js
@@ -68,16 +68,14 @@ export const withEditBlockControls = createHigherOrderComponent(
 		const { attributes, name } = props;
 		const isDisplayed = name === 'core/template-part' && attributes.slug;
 
-		if ( ! isDisplayed ) {
-			return <BlockEdit { ...props } />;
-		}
-
 		return (
 			<>
-				<BlockEdit { ...props } />
-				<BlockControls group="other">
-					<EditTemplatePartMenuItem attributes={ attributes } />
-				</BlockControls>
+				<BlockEdit key="edit" { ...props } />
+				{ isDisplayed && (
+					<BlockControls group="other">
+						<EditTemplatePartMenuItem attributes={ attributes } />
+					</BlockControls>
+				) }
 			</>
 		);
 	},