diff --git a/frontend/src/components/Diagram.jsx b/frontend/src/components/Diagram.jsx
index 27ea053..90ec8f0 100644
--- a/frontend/src/components/Diagram.jsx
+++ b/frontend/src/components/Diagram.jsx
@@ -323,12 +323,17 @@ function Diagram() {
         onNodeClick={onNodeClick}
         onNodeDoubleClick={handleNodeDoubleClick}
         onEdgeClick={onEdgeClick}
+        onPaneClick={() => {
+          setSelectedNode(null);
+          setSelectedEdge(null);
+        }}
         nodeTypes={nodeTypes}
         fitView
       >
         <Background />
         <Controls />
       </ReactFlow>
+
     </div>
   );
 }