Fix graph demo