Example not working

Added small modification to example, so it should now work.