def METHOD_NAME(self):
    self.proxy.restore_element_style()
    script = """(elements) => {
            for (element of elements) {
                element.classList.remove('rdbHighlight')
            }
        }"""
    self.browser.evaluate_javascript.assert_called_with(
        ".rdbHighlight", script, all_elements=True
    )