1 files changed, 4 insertions, 3 deletions

diff --git a/include/trace/events/kmem.h b/include/trace/events/kmem.h
index aece1346ceb7..4ad10baecd4d 100644
--- a/include/trace/events/kmem.h
+++ b/include/trace/events/kmem.h
@@ -268,11 +268,11 @@ TRACE_EVENT(mm_page_alloc_extfrag,
 
         TP_PROTO(struct page *page,
                 int alloc_order, int fallback_order,
-                int alloc_migratetype, int fallback_migratetype, int new_migratetype),
+                int alloc_migratetype, int fallback_migratetype),
 
         TP_ARGS(page,
                 alloc_order, fallback_order,
-                alloc_migratetype, fallback_migratetype, new_migratetype),
+                alloc_migratetype, fallback_migratetype),
 
         TP_STRUCT__entry(
                 __field(        struct page *,  page                    )
@@ -289,7 +289,8 @@ TRACE_EVENT(mm_page_alloc_extfrag,
                 __entry->fallback_order         = fallback_order;
                 __entry->alloc_migratetype      = alloc_migratetype;
                 __entry->fallback_migratetype   = fallback_migratetype;
-                __entry->change_ownership       = (new_migratetype == alloc_migratetype);
+                __entry->change_ownership       = (alloc_migratetype ==
+                                        get_pageblock_migratetype(page));
         ),
 
         TP_printk("page=%p pfn=%lu alloc_order=%d fallback_order=%d pageblock_order=%d alloc_migratetype=%d fallback_migratetype=%d fragmenting=%d change_ownership=%d",