#define FIRST_OBJ_PAGE_TYPE_MASK       0xffffff
 
-static inline unsigned int get_first_obj_offset(struct page *page)
+static inline unsigned int get_first_obj_offset(struct zpdesc *zpdesc)
 {
-       VM_WARN_ON_ONCE(!PageZsmalloc(page));
-       return page->page_type & FIRST_OBJ_PAGE_TYPE_MASK;
+       VM_WARN_ON_ONCE(!PageZsmalloc(zpdesc_page(zpdesc)));
+       return zpdesc->first_obj_offset & FIRST_OBJ_PAGE_TYPE_MASK;
 }
 
-static inline void set_first_obj_offset(struct page *page, unsigned int offset)
+static inline void set_first_obj_offset(struct zpdesc *zpdesc, unsigned int offset)
 {
        /* With 24 bits available, we can support offsets into 16 MiB pages. */
        BUILD_BUG_ON(PAGE_SIZE > SZ_16M);
-       VM_WARN_ON_ONCE(!PageZsmalloc(page));
+       VM_WARN_ON_ONCE(!PageZsmalloc(zpdesc_page(zpdesc)));
        VM_WARN_ON_ONCE(offset & ~FIRST_OBJ_PAGE_TYPE_MASK);
-       page->page_type &= ~FIRST_OBJ_PAGE_TYPE_MASK;
-       page->page_type |= offset & FIRST_OBJ_PAGE_TYPE_MASK;
+       zpdesc->first_obj_offset &= ~FIRST_OBJ_PAGE_TYPE_MASK;
+       zpdesc->first_obj_offset |= offset & FIRST_OBJ_PAGE_TYPE_MASK;
 }
 
 static inline unsigned int get_freeobj(struct zspage *zspage)
                struct link_free *link;
                void *vaddr;
 
-               set_first_obj_offset(zpdesc_page(zpdesc), off);
+               set_first_obj_offset(zpdesc, off);
 
                vaddr = kmap_local_zpdesc(zpdesc);
                link = (struct link_free *)vaddr + off / sizeof(*link);
        unsigned long handle = 0;
        void *addr = kmap_local_zpdesc(zpdesc);
 
-       offset = get_first_obj_offset(zpdesc_page(zpdesc));
+       offset = get_first_obj_offset(zpdesc);
        offset += class->size * index;
 
        while (offset < PAGE_SIZE) {
        } while ((zpdesc = get_next_zpdesc(zpdesc)) != NULL);
 
        create_page_chain(class, zspage, zpdescs);
-       first_obj_offset = get_first_obj_offset(zpdesc_page(oldzpdesc));
-       set_first_obj_offset(zpdesc_page(newzpdesc), first_obj_offset);
+       first_obj_offset = get_first_obj_offset(oldzpdesc);
+       set_first_obj_offset(newzpdesc, first_obj_offset);
        if (unlikely(ZsHugePage(zspage)))
                newzpdesc->handle = oldzpdesc->handle;
        __zpdesc_set_movable(newzpdesc, &zsmalloc_mops);
        /* the migrate_write_lock protects zpage access via zs_map_object */
        migrate_write_lock(zspage);
 
-       offset = get_first_obj_offset(zpdesc_page(zpdesc));
+       offset = get_first_obj_offset(zpdesc);
        s_addr = kmap_local_zpdesc(zpdesc);
 
        /*