diff --git a/resources/graphs/basic_graph/components.py b/resources/graphs/basic_graph/components.py
index cbeec67d6c8f8599f5ca409469a5967682b9fdfb..e12c82d28b3ff60f6ba9044dbf86fea395f1606e 100644
--- a/resources/graphs/basic_graph/components.py
+++ b/resources/graphs/basic_graph/components.py
@@ -211,6 +211,10 @@ class BasicEdge():
         # Get connected nodes.
         try:
             node_1, node_2 = self.get_nodes()
+
+            # Handle for dict, if classes are of Directed type.
+            if isinstance(node_1, str):
+                node_1, node_2 = self.get_nodes().values()
         except ValueError:
             # No nodes connected. Return None.
             return None