Updated the example to fix a bug