now fixing the callgraph after removing units