Fixed the example to work with the new changes