diff --git a/kernel/bpf/trampoline.c b/kernel/bpf/trampoline.c
index 9bd1731eea0dcf20a5edc9944fac0db30d46e024..e9cc7ac3856ff858def5814337eb0a838c5ea68f 100644
--- a/kernel/bpf/trampoline.c
+++ b/kernel/bpf/trampoline.c
@@ -534,11 +534,13 @@ void notrace __bpf_prog_exit(struct bpf_prog *prog, u64 start)
 void notrace __bpf_prog_enter_sleepable(void)
 {
 	rcu_read_lock_trace();
+	migrate_disable();
 	might_fault();
 }
 
 void notrace __bpf_prog_exit_sleepable(void)
 {
+	migrate_enable();
 	rcu_read_unlock_trace();
 }