Update graph tests to run correctly again (fix #37)