diff --git a/packages/e2e-tests/specs/experiments/template-part.test.js b/packages/e2e-tests/specs/experiments/template-part.test.js
index 4e16c07c4e88c..357ddc929d627 100644
--- a/packages/e2e-tests/specs/experiments/template-part.test.js
+++ b/packages/e2e-tests/specs/experiments/template-part.test.js
@@ -307,7 +307,9 @@ describe( 'Template Part', () => {
 				chooseExistingButtonSelector
 			);
 			await chooseExistingButton.click();
-			const preview = await page.waitForXPath( testContentSelector );
+			const preview = await page.waitForSelector(
+				'.block-editor-block-preview__content iframe'
+			);
 			expect( preview ).toBeTruthy();
 
 			await preview.click();