Fixing example in documentation