diff --git a/odoo-stubs/tests/common.pyi b/odoo-stubs/tests/common.pyi
index ac3b25a..dc0d281 100644
--- a/odoo-stubs/tests/common.pyi
+++ b/odoo-stubs/tests/common.pyi
@@ -1,6 +1,7 @@
 import collections
 import logging
 import unittest
+from contextlib import contextmanager
 from itertools import count
 from re import Pattern
 from typing import Any, Callable, Generator, Generic, Iterator, Mapping, Match, TypeVar
@@ -70,9 +71,11 @@ class BaseCase(TreeCase):
     def uid(self, user) -> None: ...
     def ref(self, xid: str) -> int: ...
     def browse_ref(self, xid: str) -> BaseModel | None: ...
+    @contextmanager
     def assertRaises(
         self, exception, func: Any | None = ..., *args, **kwargs
     ) -> Generator[Any, None, None] | None: ...
+    @contextmanager
     def assertQueryCount(
         self, default: int = ..., flush: bool = ..., **counters
     ) -> Generator[None, None, None]: ...