1 files changed, 3 insertions, 2 deletions

diff --git a/lib/interval_tree_test.c b/lib/interval_tree_test.c
index 835242e74aaa..75509a1511a3 100644
--- a/lib/interval_tree_test.c
+++ b/lib/interval_tree_test.c
@@ -64,11 +64,12 @@ static int interval_tree_test_init(void)
         unsigned long results;
         cycles_t time1, time2, time;
 
-        nodes = kmalloc(nnodes * sizeof(struct interval_tree_node), GFP_KERNEL);
+        nodes = kmalloc_array(nnodes, sizeof(struct interval_tree_node),
+                              GFP_KERNEL);
         if (!nodes)
                 return -ENOMEM;
 
-        queries = kmalloc(nsearches * sizeof(int), GFP_KERNEL);
+        queries = kmalloc_array(nsearches, sizeof(int), GFP_KERNEL);
         if (!queries) {
                 kfree(nodes);
                 return -ENOMEM;