clear_compound_head(page_tail);
 
        /* Finally unfreeze refcount. Additional reference from page cache. */
-       page_ref_unfreeze(page_tail, 1 + (!PageAnon(head) ||
-                                         PageSwapCache(head)));
+       page_ref_unfreeze(page_tail, 1 + (!folio_test_anon(folio) ||
+                                         folio_test_swapcache(folio)));
 
-       if (page_is_young(head))
-               set_page_young(page_tail);
-       if (page_is_idle(head))
-               set_page_idle(page_tail);
+       if (folio_test_young(folio))
+               folio_set_young(new_folio);
+       if (folio_test_idle(folio))
+               folio_set_idle(new_folio);
 
        folio_xchg_last_cpupid(new_folio, folio_last_cpupid(folio));