small fix in figure: