Move wait from page to element