1 files changed, 2 insertions, 2 deletions

diff --git a/mm/balloon_compaction.c b/mm/balloon_compaction.c
index d3116be5a00f..300117f1a08f 100644
--- a/mm/balloon_compaction.c
+++ b/mm/balloon_compaction.c
@@ -61,6 +61,7 @@ struct page *balloon_page_dequeue(struct balloon_dev_info *b_dev_info)
         bool dequeued_page;
 
         dequeued_page = false;
+        spin_lock_irqsave(&b_dev_info->pages_lock, flags);
         list_for_each_entry_safe(page, tmp, &b_dev_info->pages, lru) {
                 /*
                  * Block others from accessing the 'page' while we get around
@@ -75,15 +76,14 @@ struct page *balloon_page_dequeue(struct balloon_dev_info *b_dev_info)
                                 continue;
                         }
 #endif
-                        spin_lock_irqsave(&b_dev_info->pages_lock, flags);
                         balloon_page_delete(page);
                         __count_vm_event(BALLOON_DEFLATE);
-                        spin_unlock_irqrestore(&b_dev_info->pages_lock, flags);
                         unlock_page(page);
                         dequeued_page = true;
                         break;
                 }
         }
+        spin_unlock_irqrestore(&b_dev_info->pages_lock, flags);
 
         if (!dequeued_page) {
                 /*