diag: Update mask buffer after reallocation
am: dbeb9a83bc

Change-Id: If6a5030b812056e668998dcee63d3828412d3072