fixed figure problem