diff --git a/include/linux/mlx5/driver.h b/include/linux/mlx5/driver.h
index 32cb18c399c2048692f0c173d8708365ae849455..130bc8d77fa5e38173b90c157092189cb3ff7479 100644
--- a/include/linux/mlx5/driver.h
+++ b/include/linux/mlx5/driver.h
@@ -38,8 +38,10 @@
 #include <linux/pci.h>
 #include <linux/spinlock_types.h>
 #include <linux/semaphore.h>
+#include <linux/slab.h>
 #include <linux/vmalloc.h>
 #include <linux/radix-tree.h>
+
 #include <linux/mlx5/device.h>
 #include <linux/mlx5/doorbell.h>