diff --git a/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.component.tsx b/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.component.tsx
index 58b59da0cda..79e238c9164 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.component.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/EntityLineage/EntityLineage.component.tsx
@@ -382,19 +382,19 @@ const Entitylineage: FunctionComponent<EntityLineageProp> = ({
       ];
 
       const sourceDownstreamNode = updatedLineageData.downstreamEdges?.find(
-        (d) => source?.includes(d.toEntity as string)
+        (d) => source?.includes(d.toEntity)
       );
 
       const sourceUpStreamNode = updatedLineageData.upstreamEdges?.find((u) =>
-        source?.includes(u.fromEntity as string)
+        source?.includes(u.fromEntity)
       );
 
       const targetDownStreamNode = updatedLineageData.downstreamEdges?.find(
-        (d) => target?.includes(d.toEntity as string)
+        (d) => target?.includes(d.toEntity)
       );
 
       const targetUpStreamNode = updatedLineageData.upstreamEdges?.find((u) =>
-        target?.includes(u.fromEntity as string)
+        target?.includes(u.fromEntity)
       );
 
       let targetNode = nodes?.find((n) => target?.includes(n.id));
@@ -453,8 +453,8 @@ const Entitylineage: FunctionComponent<EntityLineageProp> = ({
           ? [
               ...(updatedLineageData.downstreamEdges as EntityEdge[]),
               {
-                fromEntity: sourceNode?.id,
-                toEntity: targetNode?.id,
+                fromEntity: sourceNode?.id as string,
+                toEntity: targetNode?.id as string,
               },
             ]
           : updatedLineageData.downstreamEdges;
@@ -468,8 +468,8 @@ const Entitylineage: FunctionComponent<EntityLineageProp> = ({
           ? [
               ...(updatedLineageData.upstreamEdges as EntityEdge[]),
               {
-                fromEntity: sourceNode?.id,
-                toEntity: targetNode?.id,
+                fromEntity: sourceNode?.id as string,
+                toEntity: targetNode?.id as string,
               },
             ]
           : updatedLineageData.upstreamEdges;