diff --git a/tests/test_awaitable.py b/tests/test_awaitable.py
index ca0b4b9..f6b2f03 100644
--- a/tests/test_awaitable.py
+++ b/tests/test_awaitable.py
@@ -52,6 +52,7 @@ def decorator(func: Callable):
             return pytest.mark.limit_leaks(memstring)(func)
         else:
             return func
+    return decorator
 
 @limit_leaks("5 KB")
 @pytest.mark.asyncio