diff --git a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java
index 511236d82..93d520eb4 100644
--- a/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java
+++ b/dhp-workflows/dhp-graph-mapper/src/test/java/eu/dnetlib/dhp/oa/graph/dump/complete/ExtractRelationFromEntityTest.java
@@ -115,6 +115,23 @@ public class ExtractRelationFromEntityTest {
 					.getReltype()
 					.getType());
 
+		Assertions
+				.assertEquals(
+						"context", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getSource().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getSource()
+								.getType());
+
+		Assertions
+				.assertEquals(
+						"result", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getSource().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getTarget()
+								.getType());
 		Assertions
 			.assertEquals(
 				"IsRelatedTo", verificationDataset
@@ -133,6 +150,23 @@ public class ExtractRelationFromEntityTest {
 					.getReltype()
 					.getType());
 
+		Assertions
+				.assertEquals(
+						"context", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getTarget().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getTarget()
+								.getType());
+
+		Assertions
+				.assertEquals(
+						"result", verificationDataset
+								.filter((FilterFunction<Relation>) row -> row.getTarget().getId().startsWith("00"))
+								.collectAsList()
+								.get(0)
+								.getSource()
+								.getType());
 	}
 
 }