Small change to the example