fix bug on figure