Flakiness fix when checking for displayed element

Made a change to combat flakiness when checking for an element to be displayed
TimeoutException I believe but I can put a webdriver exception as that is the parent of all/most selenium exceptions.