diff --git a/block/blk-wbt.c b/block/blk-wbt.c
index 4350b373a42526ba8946beee2fee0f9091618810..0d74787859e03536e2023aee953c47f8a420c06e 100644
--- a/block/blk-wbt.c
+++ b/block/blk-wbt.c
@@ -570,6 +570,7 @@ static void __wbt_wait(struct rq_wb *rwb, enum wbt_flags wb_acct,
 			io_schedule();
 
 		has_sleeper = true;
+		set_current_state(TASK_UNINTERRUPTIBLE);
 	} while (1);
 
 	finish_wait(&rqw->wait, &data.wq);