diff --git a/python/cuda/nvbench/__init__.pyi b/python/cuda/nvbench/__init__.pyi
index 80a6dee..6fea984 100644
--- a/python/cuda/nvbench/__init__.pyi
+++ b/python/cuda/nvbench/__init__.pyi
@@ -35,7 +35,7 @@ class CudaStream:
     ----
         The class is not directly constructible.
     """
-    def __cuda_stream__(self) -> tuple[int]:
+    def __cuda_stream__(self) -> tuple[int, int]:
         """
         Special method implement CUDA stream protocol
         from `cuda.core`. Returns a pair of integers: