Small fix in example and test