Update waitUntil to return the ElementProxy, and update typescript defs