Fix the example to work