diff --git a/examples/rustapi_module/tests/test_buf_and_str.py b/examples/rustapi_module/tests/test_buf_and_str.py
index 94538865..b85ad59f 100644
--- a/examples/rustapi_module/tests/test_buf_and_str.py
+++ b/examples/rustapi_module/tests/test_buf_and_str.py
@@ -34,5 +34,5 @@ def test_pybuffer_doesnot_leak_memory():
         for i in range(N):
             extractor.to_str(message_s)
 
-    memory_diff(to_vec) == 0
-    memory_diff(to_str) == 0
+    assert memory_diff(to_vec) == 0
+    assert memory_diff(to_str) == 0