Slight fix in graph output