Minor fixes for the example