Corrected missing functionality with the Selenium driver