Minor: fix code example to make it clearer