1 files changed, 1 insertions, 2 deletions

diff --git a/drivers/base/map.c b/drivers/base/map.c
index e87017f36853..c1d38234d725 100644
--- a/drivers/base/map.c
+++ b/drivers/base/map.c
@@ -41,8 +41,7 @@ int kobj_map(struct kobj_map *domain, dev_t dev, unsigned long range,
         if (n > 255)
                 n = 255;
 
-        p = kmalloc(sizeof(struct probe) * n, GFP_KERNEL);
-
+        p = kmalloc_array(n, sizeof(struct probe), GFP_KERNEL);
         if (p == NULL)
                 return -ENOMEM;