Fix graph updating