Small bug fix in example.