Fix setup code example in the docs