diff --git a/ingestion/src/metadata/ingestion/source/mlmodel/mlmodel_service.py b/ingestion/src/metadata/ingestion/source/mlmodel/mlmodel_service.py
index aa929c93295..faa07f4ecbd 100644
--- a/ingestion/src/metadata/ingestion/source/mlmodel/mlmodel_service.py
+++ b/ingestion/src/metadata/ingestion/source/mlmodel/mlmodel_service.py
@@ -148,7 +148,7 @@ class MlModelServiceSource(TopologyRunnerMixin, Source, ABC):
         self.test_connection()
         self.status = MlModelSourceStatus()
 
-        self.client = self.connection.client
+        self.client = self.connection
 
     def get_services(self) -> Iterable[WorkflowSource]:
         yield self.config