Correctly account for IO when waiting for a submitted DIO read, the case
when we're retrying.  This only for the accounting purposes and should
not change other behaviour.
Signed-off-by: David Sterba <dsterba@suse.com>
                        goto next;
                }
 
-               wait_for_completion(&done.done);
+               wait_for_completion_io(&done.done);
 
                if (!done.uptodate) {
                        /* We might have another mirror, so try again */
                        goto next;
                }
 
-               wait_for_completion(&done.done);
+               wait_for_completion_io(&done.done);
 
                if (!done.uptodate) {
                        /* We might have another mirror, so try again */