diff --git a/ingestion/src/metadata/utils/connection_clients.py b/ingestion/src/metadata/utils/connection_clients.py
index 1eb28a2dd84..115a6193e44 100644
--- a/ingestion/src/metadata/utils/connection_clients.py
+++ b/ingestion/src/metadata/utils/connection_clients.py
@@ -93,6 +93,7 @@ class DatalakeClient:
         self.config = config
 
 
+@dataclass
 class AirByteClient:
     def __init__(self, client) -> None:
         self.client = client