diff --git a/tools/slabratetop.py b/tools/slabratetop.py
index 64cecbb9c80b..f1911b094e08 100755
--- a/tools/slabratetop.py
+++ b/tools/slabratetop.py
@@ -63,7 +63,11 @@ def signal_ignore(signal, frame):
 #include <uapi/linux/ptrace.h>
 #include <linux/mm.h>
 #include <linux/slab.h>
+#ifdef CONFIG_SLUB
 #include <linux/slub_def.h>
+#else
+#include <linux/slab_def.h>
+#endif
 
 #define CACHE_NAME_SIZE 32