Fixing docblock for wait function in WebDriver