3 files changed, 6 insertions, 6 deletions

diff --git a/block/blk-exec.c b/block/blk-exec.c
index 74638ec234c8..f634de772b0c 100644
--- a/block/blk-exec.c
+++ b/block/blk-exec.c
@@ -120,9 +120,9 @@ int blk_execute_rq(struct request_queue *q, struct gendisk *bd_disk,
         /* Prevent hang_check timer from firing at us during very long I/O */
         hang_check = sysctl_hung_task_timeout_secs;
         if (hang_check)
-                while (!wait_for_completion_timeout(&wait, hang_check * (HZ/2)));
+                while (!wait_for_completion_io_timeout(&wait, hang_check * (HZ/2)));
         else
-                wait_for_completion(&wait);
+                wait_for_completion_io(&wait);
 
         if (rq->errors)
                 err = -EIO;
diff --git a/block/blk-flush.c b/block/blk-flush.c
index 720ad607ff91..db8f1b507857 100644
--- a/block/blk-flush.c
+++ b/block/blk-flush.c
@@ -436,7 +436,7 @@ int blkdev_issue_flush(struct block_device *bdev, gfp_t gfp_mask,
 
         bio_get(bio);
         submit_bio(WRITE_FLUSH, bio);
-        wait_for_completion(&wait);
+        wait_for_completion_io(&wait);
 
         /*
          * The driver must store the error location in ->bi_sector, if
diff --git a/block/blk-lib.c b/block/blk-lib.c
index b3a1f2b70b31..d6f50d572565 100644
--- a/block/blk-lib.c
+++ b/block/blk-lib.c
@@ -126,7 +126,7 @@ int blkdev_issue_discard(struct block_device *bdev, sector_t sector,
 
         /* Wait for bios in-flight */
         if (!atomic_dec_and_test(&bb.done))
-                wait_for_completion(&wait);
+                wait_for_completion_io(&wait);
 
         if (!test_bit(BIO_UPTODATE, &bb.flags))
                 ret = -EIO;
@@ -200,7 +200,7 @@ int blkdev_issue_write_same(struct block_device *bdev, sector_t sector,
 
         /* Wait for bios in-flight */
         if (!atomic_dec_and_test(&bb.done))
-                wait_for_completion(&wait);
+                wait_for_completion_io(&wait);
 
         if (!test_bit(BIO_UPTODATE, &bb.flags))
                 ret = -ENOTSUPP;
@@ -262,7 +262,7 @@ int __blkdev_issue_zeroout(struct block_device *bdev, sector_t sector,
 
         /* Wait for bios in-flight */
         if (!atomic_dec_and_test(&bb.done))
-                wait_for_completion(&wait);
+                wait_for_completion_io(&wait);
 
         if (!test_bit(BIO_UPTODATE, &bb.flags))
                 /* One of bios in the batch was completed with error.*/