Improve code in arc_buf_remove_ref