slight bug fix in example