Update demo-mode reliability

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -1464,7 +1464,14 @@ def highlight(self, selector, by=By.CSS_SELECTOR,
         loops = int(loops)
 
         o_bs = ''  # original_box_shadow
-        style = element.get_attribute('style')
+        try:
+            style = element.get_attribute('style')
+        except (StaleElementReferenceException, ENI_Exception):
+            self.wait_for_ready_state_complete()
+            time.sleep(0.05)
+            element = self.wait_for_element_visible(
+                selector, by=By.CSS_SELECTOR, timeout=settings.SMALL_TIMEOUT)
+            style = element.get_attribute('style')
         if style:
             if 'box-shadow: ' in style:
                 box_start = style.find('box-shadow: ')
@@ -3116,7 +3123,14 @@ def __highlight_with_assert_success(
             self.__slow_scroll_to_element(element)
 
         o_bs = ''  # original_box_shadow
-        style = element.get_attribute('style')
+        try:
+            style = element.get_attribute('style')
+        except (StaleElementReferenceException, ENI_Exception):
+            self.wait_for_ready_state_complete()
+            time.sleep(0.05)
+            element = self.wait_for_element_visible(
+                selector, by=By.CSS_SELECTOR, timeout=settings.SMALL_TIMEOUT)
+            style = element.get_attribute('style')
         if style:
             if 'box-shadow: ' in style:
                 box_start = style.find('box-shadow: ')