1 files changed, 3 insertions, 1 deletions

diff --git a/arch/um/os-Linux/main.c b/arch/um/os-Linux/main.c
index ea9a23696f36..fb510d40480c 100644
--- a/arch/um/os-Linux/main.c
+++ b/arch/um/os-Linux/main.c
@@ -24,6 +24,7 @@
 #include "uml-config.h"
 #include "os.h"
 #include "um_malloc.h"
+#include "kern_constants.h"
 
 /* Set in main, unchanged thereafter */
 char *linux_prog;
@@ -232,7 +233,8 @@ void *__wrap_malloc(int size)
 
         if(!CAN_KMALLOC())
                 return __real_malloc(size);
-        else if(size <= PAGE_SIZE) /* finding contiguos pages can be hard*/
+        else if(size <= UM_KERN_PAGE_SIZE)
+                /* finding contiguous pages can be hard*/
                 ret = um_kmalloc(size);
         else ret = um_vmalloc(size);