diff --git a/tests/end2end/test_end2end_tfc_w1a1.py b/tests/end2end/test_end2end_tfc_w1a1.py
index 3b8d392313d41020e6cd5728e587b57cdb281420..c30fd8dc04642be7fe59d24a2dfd10dbf29e8488 100644
--- a/tests/end2end/test_end2end_tfc_w1a1.py
+++ b/tests/end2end/test_end2end_tfc_w1a1.py
@@ -204,7 +204,7 @@ def test_end2end_tfc_w1a1_verify_dataflow_part():
     ret_rtlsim_whole = execute_onnx(model, inp_dict, True)
     res_rtlsim_whole = ret_rtlsim_whole[out_name]
     assert np.isclose(res_npysim, res_rtlsim_nodebynode).all()
-    assert np.isclose(res_rtlsim_nodebynode, res_rtlsim_whole).all()
+    assert np.isclose(res_npysim, res_rtlsim_whole).all()
 
 
 def test_end2end_tfc_w1a1_verify_all():