block: Fix blk->in_flight during blk_wait_while_drained()