diff --git a/include/linux/slab.h b/include/linux/slab.h
index 22e58c7a0158..150379855007 100644
--- a/include/linux/slab.h
+++ b/include/linux/slab.h
@@ -500,7 +500,7 @@ static __always_inline void *kmem_cache_alloc_node_trace(struct kmem_cache *s, g
 #endif /* CONFIG_TRACING */
 
 extern void *kmalloc_order(size_t size, gfp_t flags, unsigned int order) __assume_page_alignment
-									 __malloc;
+									 __alloc_size(1);
 
 #ifdef CONFIG_TRACING
 extern void *kmalloc_order_trace(size_t size, gfp_t flags, unsigned int order)