#include <nvhe/memory.h>
 #include <nvhe/spinlock.h>
 
-#define HYP_NO_ORDER   USHRT_MAX
+#define HYP_NO_ORDER   ((u8)(~0))
 
 struct hyp_pool {
        /*
        struct list_head free_area[NR_PAGE_ORDERS];
        phys_addr_t range_start;
        phys_addr_t range_end;
-       unsigned short max_order;
+       u8 max_order;
 };
 
 /* Allocation */
-void *hyp_alloc_pages(struct hyp_pool *pool, unsigned short order);
+void *hyp_alloc_pages(struct hyp_pool *pool, u8 order);
 void hyp_split_page(struct hyp_page *page);
 void hyp_get_page(struct hyp_pool *pool, void *addr);
 void hyp_put_page(struct hyp_pool *pool, void *addr);
 
  */
 static struct hyp_page *__find_buddy_nocheck(struct hyp_pool *pool,
                                             struct hyp_page *p,
-                                            unsigned short order)
+                                            u8 order)
 {
        phys_addr_t addr = hyp_page_to_phys(p);
 
 /* Find a buddy page currently available for allocation */
 static struct hyp_page *__find_buddy_avail(struct hyp_pool *pool,
                                           struct hyp_page *p,
-                                          unsigned short order)
+                                          u8 order)
 {
        struct hyp_page *buddy = __find_buddy_nocheck(pool, p, order);
 
                              struct hyp_page *p)
 {
        phys_addr_t phys = hyp_page_to_phys(p);
-       unsigned short order = p->order;
+       u8 order = p->order;
        struct hyp_page *buddy;
 
        memset(hyp_page_to_virt(p), 0, PAGE_SIZE << p->order);
 
 static struct hyp_page *__hyp_extract_page(struct hyp_pool *pool,
                                           struct hyp_page *p,
-                                          unsigned short order)
+                                          u8 order)
 {
        struct hyp_page *buddy;
 
 
 void hyp_split_page(struct hyp_page *p)
 {
-       unsigned short order = p->order;
+       u8 order = p->order;
        unsigned int i;
 
        p->order = 0;
        }
 }
 
-void *hyp_alloc_pages(struct hyp_pool *pool, unsigned short order)
+void *hyp_alloc_pages(struct hyp_pool *pool, u8 order)
 {
-       unsigned short i = order;
        struct hyp_page *p;
+       u8 i = order;
 
        hyp_spin_lock(&pool->lock);