in EditMode class there is already WebElement with this selector. You can make it proteced in EditMode class and reuse it in yours