diff --git a/include/linux/slab.h b/include/linux/slab.h
index 750b54051daf..c64bff851a8a 100644
--- a/include/linux/slab.h
+++ b/include/linux/slab.h
@@ -594,7 +594,7 @@ static __always_inline __alloc_size(1) void *kmalloc(size_t size, gfp_t flags)
 	return __kmalloc(size, flags);
 }
 
-static __always_inline void *kmalloc_node(size_t size, gfp_t flags, int node)
+static __always_inline __alloc_size(1) void *kmalloc_node(size_t size, gfp_t flags, int node)
 {
 #ifndef CONFIG_SLOB
 	if (__builtin_constant_p(size) &&