diff --git a/src/components/Graph/index.tsx b/src/components/Graph/index.tsx
index d4c81d1..4cd3e22 100644
--- a/src/components/Graph/index.tsx
+++ b/src/components/Graph/index.tsx
@@ -133,8 +133,8 @@ export const Graph = ({ isWidget = false }: { isWidget?: boolean }) => {
   const collapsedEdges = useGraph((state) => state.collapsedEdges);
 
   React.useEffect(() => {
-    const nodeList = collapsedNodes.map((id) => `[id*="node-${id}"]`);
-    const edgeList = collapsedEdges.map((id) => `[class*="edge-${id}"]`);
+    const nodeList = collapsedNodes.map((id) => `[id$="node-${id}"]`);
+    const edgeList = collapsedEdges.map((id) => `[class$="edge-${id}"]`);
 
     const hiddenItems = document.querySelectorAll(".hide");
     hiddenItems.forEach((item) => item.classList.remove("hide"));