Fixed an issue with an example