diff --git a/test/ops/embedding.py b/test/ops/embedding.py
index 99cadc1..8c00f5c 100644
--- a/test/ops/embedding.py
+++ b/test/ops/embedding.py
@@ -25,7 +25,7 @@ def test_op_embedding(
     torch_embedding(out, idx, embd)
     llaisys.Ops.embedding(out_, idx_, embd_)
 
-    check_equal(out_, out, strict=True)
+    assert check_equal(out_, out, strict=True)
 
     if profile:
         benchmark(