unsigned long *crst_table_alloc(struct mm_struct *mm)
 {
-       struct page *page = alloc_pages(GFP_KERNEL, 2);
+       struct page *page = alloc_pages(GFP_KERNEL, CRST_ALLOC_ORDER);
 
        if (!page)
                return NULL;
-       arch_set_page_dat(page, 2);
+       arch_set_page_dat(page, CRST_ALLOC_ORDER);
        return (unsigned long *) page_to_virt(page);
 }
 
 void crst_table_free(struct mm_struct *mm, unsigned long *table)
 {
-       free_pages((unsigned long) table, 2);
+       free_pages((unsigned long)table, CRST_ALLOC_ORDER);
 }
 
 static void __crst_table_upgrade(void *arg)
 
        switch (half) {
        case 0x00U:     /* pmd, pud, or p4d */
-               free_pages((unsigned long) table, 2);
+               free_pages((unsigned long)table, CRST_ALLOC_ORDER);
                return;
        case 0x01U:     /* lower 2K of a 4K page table */
        case 0x02U:     /* higher 2K of a 4K page table */