diff --git a/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceInput.pdl b/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceInput.pdl
index 1751d3514f..5b8e7ef692 100644
--- a/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceInput.pdl
+++ b/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceInput.pdl
@@ -38,7 +38,7 @@ record DataProcessInstanceInput {
   @Relationship = {
     "/*/destinationUrn": {
       "name": "DataProcessInstanceConsumes",
-      "entityTypes": [ "dataset", "mlModel" ],
+      "entityTypes": [ "dataset", "mlModel", "dataProcessInstance" ],
       "isLineage": true,
       "createdOn": "inputEdges/*/created/time"
       "createdActor": "inputEdges/*/created/actor"
diff --git a/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceOutput.pdl b/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceOutput.pdl
index 966f491e8b..438faf3e8b 100644
--- a/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceOutput.pdl
+++ b/metadata-models/src/main/pegasus/com/linkedin/dataprocess/DataProcessInstanceOutput.pdl
@@ -38,7 +38,7 @@ record DataProcessInstanceOutput {
   @Relationship = {
     "/*/destinationUrn": {
       "name": "DataProcessInstanceProduces",
-      "entityTypes": [ "dataset", "mlModel" ],
+      "entityTypes": [ "dataset", "mlModel", "dataProcessInstance" ],
       "isUpstream": false,
       "isLineage": true,
       "createdOn": "outputEdges/*/created/time"