1 files changed, 1 insertions, 1 deletions

diff --git a/lib/rbtree_test.c b/lib/rbtree_test.c
index 7d36c1e27ff6..b7055b2a07d3 100644
--- a/lib/rbtree_test.c
+++ b/lib/rbtree_test.c
@@ -247,7 +247,7 @@ static int __init rbtree_test_init(void)
         cycles_t time1, time2, time;
         struct rb_node *node;
 
-        nodes = kmalloc(nnodes * sizeof(*nodes), GFP_KERNEL);
+        nodes = kmalloc_array(nnodes, sizeof(*nodes), GFP_KERNEL);
         if (!nodes)
                 return -ENOMEM;