Fix fig 2 (#21)