Fixed test.
Added getGraph() support to the graph handler.