Use `arg` instead of `node`, otherwise the highlighted range will be too big.