Merge request in regards to issue #326

I'm not fully sure why there is an error screenshot when WebDriver doesn't implement that model. This has fixed the issue I highlighted for now.