fix code example in docs.