diff --git a/pathfinding/core/graph.py b/pathfinding/core/graph.py
index 1226edd..0acf95a 100644
--- a/pathfinding/core/graph.py
+++ b/pathfinding/core/graph.py
@@ -4,7 +4,7 @@
 
 class Graph:
     def __init__(
-            self, edges: List[Set] = None, nodes: Dict[int, GraphNode] = None,
+            self, edges: List[List] = None, nodes: Dict[int, GraphNode] = None,
             bi_directional: bool = False):
         # edges defined by node-from, node-to and its cost
         self.edges = edges if edges else []