Forgot to fixup one code example in the docs