diff --git a/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py b/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py
index 92e2d0200c80d4f9ce866757c62e9617cd9fdcf3..e10b20a073aec0b8b15f40917073035ebcfc84a2 100644
--- a/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py
+++ b/python/paddle/fluid/tests/unittests/test_sparse_attention_op.py
@@ -405,7 +405,7 @@ class TestSparseAttentionAPI(unittest.TestCase):
                 )
 
             np.testing.assert_allclose(
-                fetches_result, expected_result, rtol=1e-05, atol=1e-05
+                fetches_result[0], expected_result, rtol=1e-05, atol=1e-05
             )
 
     def test_dygraph(self):