diff --git a/stdlib/asyncio/tasks.pyi b/stdlib/asyncio/tasks.pyi
index 9da1d92f5..65f61b887 100644
--- a/stdlib/asyncio/tasks.pyi
+++ b/stdlib/asyncio/tasks.pyi
@@ -149,9 +149,16 @@ def gather(
 def run_coroutine_threadsafe(coro: _FutureT[_T], loop: AbstractEventLoop) -> concurrent.futures.Future[_T]: ...
 def shield(arg: _FutureT[_T], *, loop: Optional[AbstractEventLoop] = ...) -> Future[_T]: ...
 def sleep(delay: float, result: _T = ..., *, loop: Optional[AbstractEventLoop] = ...) -> Future[_T]: ...
+@overload
+def wait(fs: Iterable[_FT], *, loop: Optional[AbstractEventLoop] = ..., timeout: Optional[float] = ..., return_when: str = ...) -> Future[Tuple[Set[_FT], Set[_FT]]]: ...  # type: ignore
+@overload
 def wait(
-    fs: Iterable[_FutureT[_T]], *, loop: Optional[AbstractEventLoop] = ..., timeout: Optional[float] = ..., return_when: str = ...
-) -> Future[Tuple[Set[Future[_T]], Set[Future[_T]]]]: ...
+    fs: Iterable[Awaitable[_T]],
+    *,
+    loop: Optional[AbstractEventLoop] = ...,
+    timeout: Optional[float] = ...,
+    return_when: str = ...,
+) -> Future[Tuple[Set[Task[_T]], Set[Task[_T]]]]: ...
 def wait_for(fut: _FutureT[_T], timeout: Optional[float], *, loop: Optional[AbstractEventLoop] = ...) -> Future[_T]: ...
 
 class Task(Future[_T], Generic[_T]):