I'd recommend using following approach for timeout changing (haven't tested myself, but looking at code it should work well):

```php
$this->wdSession->timeouts($type, $param);
```

Also support for `page load` timeout (how much to wait before considering, that the page is loaded) must be added as well.