fix bug on the example