fixed figure issue