diff --git a/drivers/ub/urma/uburma/uburma_main.c b/drivers/ub/urma/uburma/uburma_main.c
index b2e11fe5344cdce4c07c46578ff16867bb50ad5b..67871364e2063c0867b2975cd44c88049f0bf506 100644
--- a/drivers/ub/urma/uburma/uburma_main.c
+++ b/drivers/ub/urma/uburma/uburma_main.c
@@ -18,6 +18,7 @@
 #include <linux/inet.h>
 #include <linux/netdevice.h>
 #include <linux/version.h>
+#include <linux/huge_mm.h>
 
 #include "ub/urma/ubcore_types.h"
 #include "ub/urma/ubcore_uapi.h"
@@ -63,13 +64,13 @@ static struct class g_uburma_class = {
 
 static const struct file_operations g_uburma_fops = {
 	.owner = THIS_MODULE,
-	// .write	 = uburma_write,
 	.mmap = uburma_mmap,
 	.open = uburma_open,
 	.release = uburma_close,
 	.llseek = no_llseek,
 	.unlocked_ioctl = uburma_ioctl,
 	.compat_ioctl = uburma_ioctl,
+	.get_unmapped_area = thp_get_unmapped_area,
 };
 
 static int uburma_add_device(struct ubcore_device *ubc_dev);