diff --git a/stubs/docker/docker/models/containers.pyi b/stubs/docker/docker/models/containers.pyi
index 3f3b0f17f..763ebb9f5 100644
--- a/stubs/docker/docker/models/containers.pyi
+++ b/stubs/docker/docker/models/containers.pyi
@@ -1,5 +1,5 @@
 from _typeshed import Incomplete
-from typing import NamedTuple
+from typing import Literal, NamedTuple
 
 from docker.types.daemon import CancellableStream
 
@@ -57,7 +57,7 @@ class Container(Model):
     def top(self, **kwargs): ...
     def unpause(self): ...
     def update(self, **kwargs): ...
-    def wait(self, **kwargs): ...
+    def wait(self, *, timeout: float | None = None, condition: Literal["not-running", "next-exit", "removed"] | None = None): ...
 
 class ContainerCollection(Collection[Container]):
     model: type[Container]