Fixed example to make it work