Fix test that fails because the graph/diagram is non-deterministic