diff --git a/mm/vmalloc.c b/mm/vmalloc.c
index e03f4c7307a5b2dc4d7535815c69a786b2124f5d..7e25ee3ce6e5485649ba96f027b184357a2919f8 100644
--- a/mm/vmalloc.c
+++ b/mm/vmalloc.c
@@ -413,11 +413,11 @@ static struct vmap_area *alloc_vmap_area(unsigned long size,
 		if (addr + size - 1 < addr)
 			goto overflow;
 
-		n = rb_next(&first->rb_node);
-		if (n)
-			first = rb_entry(n, struct vmap_area, rb_node);
-		else
+		if (list_is_last(&first->list, &vmap_area_list))
 			goto found;
+
+		first = list_entry(first->list.next,
+				struct vmap_area, list);
 	}
 
 found: