diff --git a/lib/linux_compat.c b/lib/linux_compat.c
index a936a7e..6373b44 100644
--- a/lib/linux_compat.c
+++ b/lib/linux_compat.c
@@ -1,5 +1,6 @@
 
 #include <common.h>
+#include <memalign.h>
 #include <linux/compat.h>
 
 struct p_current cur = {
@@ -18,7 +19,7 @@ void *kmalloc(size_t size, int flags)
 {
 	void *p;
 
-	p = memalign(ARCH_DMA_MINALIGN, size);
+	p = malloc_cache_aligned(size);
 	if (flags & __GFP_ZERO)
 		memset(p, 0, size);
 
@@ -37,5 +38,5 @@ struct kmem_cache *get_mem(int element_sz)
 
 void *kmem_cache_alloc(struct kmem_cache *obj, int flag)
 {
-	return memalign(ARCH_DMA_MINALIGN, obj->sz);
+	return malloc_cache_aligned(obj->sz);
 }