diff --git a/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py b/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py
index 6169a6fc2af..9e2663d435b 100644
--- a/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py
+++ b/ingestion/src/metadata/ingestion/ometa/mixins/mlmodel_mixin.py
@@ -79,10 +79,10 @@ class OMetaMlModelMixin(OMetaLineageMixin):
                 AddLineageRequest(
                     description="MlModel uses FeatureSource",
                     edge=EntitiesEdge(
-                        fromEntity=self.get_entity_reference(
+                        fromEntity=entity_ref,
+                        toEntity=self.get_entity_reference(
                             entity=MlModel, fqn=model.fullyQualifiedName
                         ),
-                        toEntity=entity_ref,
                     ),
                 )
             )