diff --git a/ingestion/tests/cli_e2e/base/config_builders/builders.py b/ingestion/tests/cli_e2e/base/config_builders/builders.py
index 3368fcb6ed0..a1a213eba21 100644
--- a/ingestion/tests/cli_e2e/base/config_builders/builders.py
+++ b/ingestion/tests/cli_e2e/base/config_builders/builders.py
@@ -139,14 +139,20 @@ class DataQualityConfigBuilder(BaseBuilder):
 
     def build(self) -> dict:
         """build profiler config"""
-        del self.config["source"]["sourceConfig"]["config"]
-        self.config["source"]["sourceConfig"] = {
-            "config": {
-                "type": TestSuiteConfigType.TestSuite.value,
-                "entityFullyQualifiedName": self.entity_fqn,
-            },
+        self.config["source"]["sourceConfig"]["config"] = {
+            "type": TestSuiteConfigType.TestSuite.value,
+            "entityFullyQualifiedName": self.entity_fqn,
         }
 
+        self.config["source"]["sourceConfig"]["config"]["serviceConnections"] = [
+            {
+                "serviceName": self.config["source"]["serviceName"],
+                "serviceConnection": self.config["source"]["serviceConnection"],
+            }
+        ]
+
+        del self.config["source"]["serviceConnection"]
+
         self.config["processor"] = {
             "type": "orm-test-runner",
             "config": {"testCases": self.test_case_defintions},