diff --git a/block/blk-mq.c b/block/blk-mq.c
index df93102e21494dc7f98456e4376ce7c83fca56ab..357492712b0ea85362d2bbd6b2aa1687459cba66 100644
--- a/block/blk-mq.c
+++ b/block/blk-mq.c
@@ -3164,6 +3164,7 @@ static bool __blk_mq_poll(struct blk_mq_hw_ctx *hctx, struct request *rq)
 		cpu_relax();
 	}
 
+	__set_current_state(TASK_RUNNING);
 	return false;
 }