Fix graph display bug