From 7729cb63a7872a22c6b1886a3f8f340b96960430 Mon Sep 17 00:00:00 2001
From: isabelrios <isabelrios@gmail.com>
Date: Wed, 14 Aug 2019 18:57:36 +0200
Subject: [PATCH] Fix UI Screenshot TabMenuTest (#4730)

---
 .../java/org/mozilla/fenix/ui/screenshots/MenuScreenShotTest.kt  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/screenshots/MenuScreenShotTest.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/screenshots/MenuScreenShotTest.kt
index e09e76990..839cf93c8 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/screenshots/MenuScreenShotTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/screenshots/MenuScreenShotTest.kt
@@ -209,6 +209,7 @@ class MenuScreenShotTest : ScreenshotTest() {
         device.findObject(UiSelector()
                 .instance(1)
                 .className(ImageView::class.java)).click()
+        device.wait(Until.findObjects(By.res("close_tab_button")), TestAssetHelper.waitingTimeShort)
         closeTabButton()
         Screengrab.screenshot("remove-tab")
     }