diff --git a/kernel/audit.c b/kernel/audit.c
index 993e445..b86007d 100644
--- a/kernel/audit.c
+++ b/kernel/audit.c
@@ -611,13 +611,13 @@
 	spin_unlock_irqrestore(&audit_freelist_lock, flags);
 
 	if (!ab) {
-		ab = kmalloc(sizeof(*ab), GFP_ATOMIC);
+		ab = kmalloc(sizeof(*ab), gfp_mask);
 		if (!ab)
 			goto err;
 	}
 	atomic_inc(&audit_backlog);
 
-	ab->skb = alloc_skb(AUDIT_BUFSIZ, GFP_ATOMIC);
+	ab->skb = alloc_skb(AUDIT_BUFSIZ, gfp_mask);
 	if (!ab->skb)
 		goto err;