* Use public API to add argument to the selenium driver. While `args` is also public API it doesn't know how to handle if no argument is being set yet in Selenium 4. * Fix assertions to match return on Selenium 4