Split the op setting code into a helper, use it in both places.
Signed-off-by: NJens Axboe <axboe@fb.com>