Let's make the example work as expected