1 files changed, 2 insertions, 1 deletions

diff --git a/fs/nilfs2/inode.c b/fs/nilfs2/inode.c
index 1c0e8fedc095..7aaf913e8709 100644
--- a/fs/nilfs2/inode.c
+++ b/fs/nilfs2/inode.c
@@ -304,6 +304,7 @@ nilfs_direct_IO(int rw, struct kiocb *iocb, struct iov_iter *iter,
         struct file *file = iocb->ki_filp;
         struct address_space *mapping = file->f_mapping;
         struct inode *inode = file->f_mapping->host;
+        size_t count = iov_iter_count(iter);
         ssize_t size;
 
         if (rw == WRITE)
@@ -319,7 +320,7 @@ nilfs_direct_IO(int rw, struct kiocb *iocb, struct iov_iter *iter,
          */
         if (unlikely((rw & WRITE) && size < 0)) {
                 loff_t isize = i_size_read(inode);
-                loff_t end = offset + iov_length(iter->iov, iter->nr_segs);
+                loff_t end = offset + count;
 
                 if (end > isize)
                         nilfs_write_failed(mapping, end);