diff --git a/neo4j_queries/edge_queries.py b/neo4j_queries/edge_queries.py
index 02d81287d495bbbf945036e6173471a46b16a347..c3970092e8e98489fa2f51a6ada22bbff145997e 100644
--- a/neo4j_queries/edge_queries.py
+++ b/neo4j_queries/edge_queries.py
@@ -71,7 +71,7 @@ def create_data_relationship(driver: Driver, from_internal_node_id: int, to_inte
     to_internal_node_id (int): the internal Neo4j ID of the second node
 
     Returns:
-    tuple[str,str]: from_internal_node_id, to_internal_node_id
+    tuple[int,int]: from_internal_node_id, to_internal_node_id
     """
     query = """
     MATCH (a), (b)