Fix: AI - Screenshots are not being generated properly

matipojo · Nov 14, 2023 · 3bc2f1a · 3bc2f1a
1 parent 5bd0275
commit 3bc2f1a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/modules/ai/assets/js/editor/utils/screenshot.js b/modules/ai/assets/js/editor/utils/screenshot.js
@@ -20,6 +20,10 @@ export const takeScreenshot = async ( template ) => {
 	// Wait for the container to render.
 	await waitForContainer( container.id );
 
+	if ( template.elements.length ) {
+		await Promise.all( template.elements.map( ( child ) => waitForContainer( child.id ) ) );
+	}
+
 	let screenshot;
 
 	try {