diff --git a/mm/share_pool.c b/mm/share_pool.c
index 400a702b30eb50fd50de4350e186f4c85e62dfb9..ef74de39053b233948e60f48ef25cc568d41b5e1 100644
--- a/mm/share_pool.c
+++ b/mm/share_pool.c
@@ -2856,10 +2856,11 @@ static int sp_k2u_prepare(unsigned long kva, unsigned long size,
 
 	trace_sp_k2u_begin(kc);
 
-	if (sp_flags & ~SP_DVPP) {
+	if (sp_flags & ~SP_FLAG_MASK) {
 		pr_err_ratelimited("k2u sp_flags %lx error\n", sp_flags);
 		return -EINVAL;
 	}
+	sp_flags &= ~SP_HUGEPAGE;
 
 	if (!current->mm) {
 		pr_err_ratelimited("k2u: kthread is not allowed\n");