diff --git a/src/container/EdgeRenderer/index.tsx b/src/container/EdgeRenderer/index.tsx
index 8f11bac9..aa42964b 100644
--- a/src/container/EdgeRenderer/index.tsx
+++ b/src/container/EdgeRenderer/index.tsx
@@ -77,7 +77,8 @@ const Edge = ({
     return null;
   }
 
-  if (!sourceNode.__rf.width || !sourceNode.__rf.height) {
+  // source and target node need to be initialized
+  if (!sourceNode.__rf.width || !targetNode.__rf.width) {
     return null;
   }