Fix graph tests