Как наследовать от WebElement в WebdriverIO

Question

Я искал способ унаследовать от объекта WebElement, который возвращает webdriverio, без применения мартышек и с поддержкой типов TS (автозаполнение является обязательным). Есть ли способ сделать что-то подобное?

class Checkbox extends WebdriverIOWebElement {
    constructor() {
       super($('div'))
    }
    // overriding base method
    isDisplayed(): boolean {
        // blabla some new logic here
    }

    check() {
        if(!this.isChecked()) {
            this.click()
        }
    }

    uncheck() {
        if(this.isChecked()) {
            this.click()
        }
    }
}