return 0;
 }
 
-static bool zswap_compress(struct folio *folio, struct zswap_entry *entry)
+static bool zswap_compress(struct page *page, struct zswap_entry *entry)
 {
        struct crypto_acomp_ctx *acomp_ctx;
        struct scatterlist input, output;
 
        dst = acomp_ctx->buffer;
        sg_init_table(&input, 1);
-       sg_set_folio(&input, folio, PAGE_SIZE, 0);
+       sg_set_page(&input, page, PAGE_SIZE, 0);
 
        /*
         * We need PAGE_SIZE * 2 here since there maybe over-compression case,
                mem_cgroup_put(memcg);
        }
 
-       if (!zswap_compress(folio, entry))
+       if (!zswap_compress(&folio->page, entry))
                goto put_pool;
 
        entry->swpentry = swp;