Fix to work the example code