Fix output handling logic in graph.rs

Fix output handling logic in graph.rs #6315