diff --git a/tests/end_to_end/conftest.py b/tests/end_to_end/conftest.py
index bfcdd2fd47..d2c9c20f89 100644
--- a/tests/end_to_end/conftest.py
+++ b/tests/end_to_end/conftest.py
@@ -224,6 +224,7 @@ def fx_federation(request, pytestconfig):
     """
     collaborators = []
     agg_domain_name = "localhost"
+
     # Parse the command line arguments
     args = parse_arguments()
     # Use the model name from the test case name if not provided as a command line argument
diff --git a/tests/end_to_end/test_suites/task_runner_tests.py b/tests/end_to_end/test_suites/task_runner_tests.py
index 6327ee6301..371fee8f08 100644
--- a/tests/end_to_end/test_suites/task_runner_tests.py
+++ b/tests/end_to_end/test_suites/task_runner_tests.py
@@ -16,15 +16,15 @@ def test_torch_cnn_mnist(fx_federation):
     """
     log.info("Testing torch_cnn_mnist model")
 
-    # # Setup PKI for trusted communication within the federation
-    # if not fx_federation.disable_tls:
-    #     assert fed_helper.setup_pki(fx_federation), "Failed to setup PKI for trusted communication"
+    # Setup PKI for trusted communication within the federation
+    if not fx_federation.disable_tls:
+        assert fed_helper.setup_pki(fx_federation), "Failed to setup PKI for trusted communication"
 
-    # # Start the federation
-    # results = fed_helper.run_federation(fx_federation)
+    # Start the federation
+    results = fed_helper.run_federation(fx_federation)
 
-    # # Verify the completion of the federation run
-    # assert fed_helper.verify_federation_run_completion(fx_federation, results), "Federation completion failed"
+    # Verify the completion of the federation run
+    assert fed_helper.verify_federation_run_completion(fx_federation, results), "Federation completion failed"
 
 
 @pytest.mark.keras_cnn_mnist
diff --git a/tests/end_to_end/utils/summary_helper.py b/tests/end_to_end/utils/summary_helper.py
index 6bb9256cc5..599c4b6ec5 100644
--- a/tests/end_to_end/utils/summary_helper.py
+++ b/tests/end_to_end/utils/summary_helper.py
@@ -107,6 +107,7 @@ def main():
     """
     Main function to get the test case results and aggregator logs
     And write the results to GitHub step summary
+    IMP: Do not fail the test in any scenario
     """
     result = get_testcase_result()