diff --git a/test/conformance/conformance_graph.cpp b/test/conformance/conformance_graph.cpp
index 7c6af23707e..8976cd10ad1 100644
--- a/test/conformance/conformance_graph.cpp
+++ b/test/conformance/conformance_graph.cpp
@@ -76,7 +76,6 @@ void test_functional_nodes_rf_reset_protocol(){
 
     f.try_put(0);
     f.try_put(0);
-
     CHECK_MESSAGE((counting_body.execute_count == 0), "Body should not be executed");
     g.reset(oneapi::tbb::flow::rf_reset_protocol);
 
@@ -94,7 +93,7 @@ void test_buffering_nodes_rf_reset_protocol(Args... node_body){
     CHECK_MESSAGE((tmp == -1), "Value should not be updated");
 
     testing_node.try_put(1);
-
+    g.wait_for_all();
     g.reset(oneapi::tbb::flow::rf_reset_protocol);
 
     tmp = -1;