return zram_read_page(zram, bvec->bv_page, index, bio, false);
 }
 
-static int __zram_bvec_write(struct zram *zram, struct bio_vec *bvec,
-                               u32 index, struct bio *bio)
+static int zram_write_page(struct zram *zram, struct page *page, u32 index)
 {
        int ret = 0;
        unsigned long alloced_pages;
        unsigned int comp_len = 0;
        void *src, *dst, *mem;
        struct zcomp_strm *zstrm;
-       struct page *page = bvec->bv_page;
        unsigned long element = 0;
        enum zram_pageflags flags = 0;
 
 static int zram_bvec_write(struct zram *zram, struct bio_vec *bvec,
                                u32 index, int offset, struct bio *bio)
 {
+       struct page *page = bvec->bv_page;
        int ret;
-       struct page *page = NULL;
-       struct bio_vec vec;
 
-       vec = *bvec;
        if (is_partial_io(bvec)) {
                /*
                 * This is a partial IO. We need to read the full page
                        goto out;
 
                memcpy_from_bvec(page_address(page) + offset, bvec);
-
-               bvec_set_page(&vec, page, PAGE_SIZE, 0);
        }
 
-       ret = __zram_bvec_write(zram, &vec, index, bio);
+       ret = zram_write_page(zram, page, index);
 out:
        if (is_partial_io(bvec))
                __free_page(page);
 
        /*
         * No direct reclaim (slow path) for handle allocation and no
-        * re-compression attempt (unlike in __zram_bvec_write()) since
+        * re-compression attempt (unlike in zram_write_bvec()) since
         * we already have stored that object in zsmalloc. If we cannot
         * alloc memory for recompressed object then we bail out and
         * simply keep the old (existing) object in zsmalloc.