Fix a bug in the example.