diff --git a/test/helpers/test_leetcode.py b/test/helpers/test_leetcode.py
index a0e7e46..40b9e34 100644
--- a/test/helpers/test_leetcode.py
+++ b/test/helpers/test_leetcode.py
@@ -79,9 +79,9 @@ async def test() -> str:
 
         wrapper = decorator(func)
 
-        await wrapper() == "test"
+        assert (await wrapper()) == "test"
 
-        func.call_count = 3
+        assert func.call_count == 3
 
 
 class TestLeetcodeData: