How to avoid type casting while waiting for mobile element?

Question

As a main type I'm using MobileElement, however I need to wait for element.

For explicit wait I'm using:

WebDriverWait wait = new WebDriverWait(driver, 2);

While my Android driver accepts only MobileELement the Expected conditions library requires type casting because of incompatibility of types.

public MobileElement getElement(By id) {
    return (MobileElement) wait.until(ExpectedConditions.visibilityOfElementLocated(id));
}

Is there any solution to avoid type casting or other library for wait purposes?