Update to code because of the selenium bug fix.