Fix graph re-rendering