1 files changed, 1 insertions, 1 deletions
diff --git a/plugins/SegmentEditor/tests/UI/SegmentSelectorEditor_spec.js b/plugins/SegmentEditor/tests/UI/SegmentSelectorEditor_spec.js
index 35179aa6eb..ad988111e2 100644
--- a/plugins/SegmentEditor/tests/UI/SegmentSelectorEditor_spec.js
+++ b/plugins/SegmentEditor/tests/UI/SegmentSelectorEditor_spec.js
@@ -188,7 +188,7 @@ describe("SegmentSelectorEditorTest", function () {
         var elem = await page.jQuery('.modal.open .modal-footer a:contains(Yes):visible');
         await elem.click();
         await page.waitForNetworkIdle();
-        await page.click('.segmentationContainer');
+        await (await page.waitForSelector('.segmentationContainer')).click();
         await page.waitForNetworkIdle();
         expect(await page.screenshotSelector(selectorsToCapture)).to.matchImage('updated');
     });