1 files changed, 3 insertions, 2 deletions

diff --git a/lib/test_kmod.c b/lib/test_kmod.c
index 0e5b7a61460b..e3ddd836491f 100644
--- a/lib/test_kmod.c
+++ b/lib/test_kmod.c
@@ -779,8 +779,9 @@ static int kmod_config_sync_info(struct kmod_test_device *test_dev)
         struct test_config *config = &test_dev->config;
 
         free_test_dev_info(test_dev);
-        test_dev->info = vzalloc(config->num_threads *
-                                 sizeof(struct kmod_test_device_info));
+        test_dev->info =
+                vzalloc(array_size(sizeof(struct kmod_test_device_info),
+                                   config->num_threads));
         if (!test_dev->info)
                 return -ENOMEM;