bug fix
modifed the example