diff --git a/block/blk-wbt.c b/block/blk-wbt.c
index f92fc84b5e2c497b77578408c10c22bf37929384..3e34b41bcefc906cf54948e7c683da03a586223f 100644
--- a/block/blk-wbt.c
+++ b/block/blk-wbt.c
@@ -701,7 +701,7 @@ static int wbt_data_dir(const struct request *rq)
 
 	if (op == REQ_OP_READ)
 		return READ;
-	else if (op == REQ_OP_WRITE || op == REQ_OP_FLUSH)
+	else if (op_is_write(op))
 		return WRITE;
 
 	/* don't account */