fixing code example in docs