diff --git a/mailpoet/tests/performance/tests/segments-select-template.js b/mailpoet/tests/performance/tests/segments-select-template.js
index 00b96af89f..51c996ce99 100644
--- a/mailpoet/tests/performance/tests/segments-select-template.js
+++ b/mailpoet/tests/performance/tests/segments-select-template.js
@@ -51,13 +51,15 @@ export async function segmentsSelectTemplate() {
       page.waitForLoadState('networkidle'),
     ]);
 
+    // Save the segment
+    await focusAndClick(page, 'button[type="submit"]');
+
+    await page.waitForLoadState('networkidle');
     await page.screenshot({
       path: screenshotPath + 'Segments_Select_Template_02.png',
       fullPage: fullPageSet,
     });
 
-    // Save the segment
-    await focusAndClick(page, 'button[type="submit"]');
     await page.waitForSelector('[data-automation-id="filters_all"]', {
       state: 'visible',
     });