diff --git a/src/Search/Search.hs b/src/Search/Search.hs
index 9fcbf3af0c97a2bf2cfa888e3ead2b371642c17d..9b1e012ebafc1ec6e27f39c4e3c7efa413f54156 100644
--- a/src/Search/Search.hs
+++ b/src/Search/Search.hs
@@ -16,6 +16,7 @@ import Control.Monad
 import System.Exit
 import System.Process
 import System.IO
+import Data.Foldable
 
 -- | (Syscall, Statehash)
 type Vertex = (Text, Text)
@@ -156,8 +157,9 @@ graph2Dot graph =
       T.drop (T.length "OSEKOS_") txt
 
 dotNodes :: Graph -> Text
-dotNodes = foldMap formatNode . M.keys
+dotNodes = foldMap formatNode . allNotes
   where
+    allNotes m = M.keysSet m `S.union` (fold m)
     formatNode (call, state) = nodeId (call,state)
                             <> " [label=\""
                             <> "State: " <> state <> "\\n"