diff --git a/tests/icosagon/test_input.py b/tests/icosagon/test_input.py
index 4f50912..8bb5dc0 100644
--- a/tests/icosagon/test_input.py
+++ b/tests/icosagon/test_input.py
@@ -10,7 +10,7 @@ def _some_data():
     d.add_node_type('Gene', 1000)
     d.add_node_type('Drug', 100)
 
-    fam = d.add_relation_family('Drug-Gene', 1, 0, False)
+    fam = d.add_relation_family('Drug-Gene', 1, 0, True)
     fam.add_relation_type('Target', torch.rand(100, 1000))
 
     fam = d.add_relation_family('Gene-Gene', 0, 0, False)