Update example in docs, fixes #356