diff --git a/openmetadata-docs/content/v1.5.x/sdk/python/ingestion/lineage.md b/openmetadata-docs/content/v1.5.x/sdk/python/ingestion/lineage.md
index 59b15e969ce1..8088acb9ac84 100644
--- a/openmetadata-docs/content/v1.5.x/sdk/python/ingestion/lineage.md
+++ b/openmetadata-docs/content/v1.5.x/sdk/python/ingestion/lineage.md
@@ -166,6 +166,7 @@ from metadata.generated.schema.type.entityLineage import EntitiesEdge
 add_lineage_request = AddLineageRequest(
     edge=EntitiesEdge(
         description="test lineage",
+        from metadata.generated.schema.type.entityReference import EntityReference
         fromEntity=EntityReference(id=table_a_entity.id, type="table"),
         toEntity=EntityReference(id=table_b_entity.id, type="table"),
     ),
@@ -325,6 +326,7 @@ lineage_details = LineageDetails(
 
 add_lineage_request = AddLineageRequest(
     edge=EntitiesEdge(
+        from metadata.generated.schema.type.entityReference import EntityReference
         fromEntity=EntityReference(id=table_a_entity.id, type="table"),
         toEntity=EntityReference(id=table_c_entity.id, type="table"),
         lineageDetails=lineage_details,
diff --git a/openmetadata-docs/content/v1.6.x-SNAPSHOT/sdk/python/ingestion/lineage.md b/openmetadata-docs/content/v1.6.x-SNAPSHOT/sdk/python/ingestion/lineage.md
index 59b15e969ce1..8088acb9ac84 100644
--- a/openmetadata-docs/content/v1.6.x-SNAPSHOT/sdk/python/ingestion/lineage.md
+++ b/openmetadata-docs/content/v1.6.x-SNAPSHOT/sdk/python/ingestion/lineage.md
@@ -166,6 +166,7 @@ from metadata.generated.schema.type.entityLineage import EntitiesEdge
 add_lineage_request = AddLineageRequest(
     edge=EntitiesEdge(
         description="test lineage",
+        from metadata.generated.schema.type.entityReference import EntityReference
         fromEntity=EntityReference(id=table_a_entity.id, type="table"),
         toEntity=EntityReference(id=table_b_entity.id, type="table"),
     ),
@@ -325,6 +326,7 @@ lineage_details = LineageDetails(
 
 add_lineage_request = AddLineageRequest(
     edge=EntitiesEdge(
+        from metadata.generated.schema.type.entityReference import EntityReference
         fromEntity=EntityReference(id=table_a_entity.id, type="table"),
         toEntity=EntityReference(id=table_c_entity.id, type="table"),
         lineageDetails=lineage_details,