1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/xen/balloon.c b/drivers/xen/balloon.c
index 39b229f9e256..d37dd5bb7a8f 100644
--- a/drivers/xen/balloon.c
+++ b/drivers/xen/balloon.c
@@ -604,6 +604,7 @@ int alloc_xenballooned_pages(int nr_pages, struct page **pages)
         while (pgno < nr_pages) {
                 page = balloon_retrieve(true);
                 if (page) {
+                        __ClearPageOffline(page);
                         pages[pgno++] = page;
 #ifdef CONFIG_XEN_HAVE_PVMMU
                         /*
@@ -645,8 +646,10 @@ void free_xenballooned_pages(int nr_pages, struct page **pages)
         mutex_lock(&balloon_mutex);
 
         for (i = 0; i < nr_pages; i++) {
-                if (pages[i])
+                if (pages[i]) {
+                        __SetPageOffline(pages[i]);
                         balloon_append(pages[i]);
+                }
         }
 
         balloon_stats.target_unpopulated -= nr_pages;