diff --git a/subiquity/tests/api/test_api.py b/subiquity/tests/api/test_api.py
index 911d6756..f1f03870 100755
--- a/subiquity/tests/api/test_api.py
+++ b/subiquity/tests/api/test_api.py
@@ -36,7 +36,7 @@ def timeout(multiplier=1):
     def wrapper(coro):
         @wraps(coro)
         async def run(*args, **kwargs):
-            with async_timeout.timeout(default_timeout * multiplier):
+            async with async_timeout.timeout(default_timeout * multiplier):
                 return await coro(*args, **kwargs)
         return run
     return wrapper