Codeception
diff --git a/‎module/VisualCeption.php‎
Lines changed: 6 additions & 2 deletions b/‎module/VisualCeption.php‎
Lines changed: 6 additions & 2 deletions
@@ -380,7 +380,9 @@ private function hideElementsForScreenshot(array $excludeElements)
         foreach ($excludeElements as $element) {
             $this->hideElement($element);
         }
-        $this->webDriverModule->wait(1);
+        if (!empty($excludeElements)) {
+            $this->webDriverModule->waitForElementNotVisible(array_pop($excludeElements));
+        }
     }
 
     /**
@@ -393,7 +395,9 @@ private function resetHideElementsForScreenshot(array $excludeElements)
         foreach ($excludeElements as $element) {
             $this->showElement($element);
         }
-        $this->webDriverModule->wait(1);
+        if (!empty($excludeElements)) {
+            $this->webDriverModule->waitForElementVisible(array_pop($excludeElements));
+        }
     }
 
     /**
Original file line number	Diff line number	Diff line change
`@@ -380,7 +380,9 @@ private function hideElementsForScreenshot(array $excludeElements)`
`380`	`380`	`foreach ($excludeElements as $element) {`
`381`	`381`	`$this->hideElement($element);`
`382`	`382`	`}`
`383`		`- $this->webDriverModule->wait(1);`
	`383`	`+ if (!empty($excludeElements)) {`
	`384`	`+ $this->webDriverModule->waitForElementNotVisible(array_pop($excludeElements));`
	`385`	`+ }`
`384`	`386`	`}`
`385`	`387`
`386`	`388`	`/**`
`@@ -393,7 +395,9 @@ private function resetHideElementsForScreenshot(array $excludeElements)`
`393`	`395`	`foreach ($excludeElements as $element) {`
`394`	`396`	`$this->showElement($element);`
`395`	`397`	`}`
`396`		`- $this->webDriverModule->wait(1);`
	`398`	`+ if (!empty($excludeElements)) {`
	`399`	`+ $this->webDriverModule->waitForElementVisible(array_pop($excludeElements));`
	`400`	`+ }`
`397`	`401`	`}`
`398`	`402`
`399`	`403`	`/**`