1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/base/node.c b/drivers/base/node.c
index 92b00a7e6a02..7a3a580821e0 100644
--- a/drivers/base/node.c
+++ b/drivers/base/node.c
@@ -407,6 +407,8 @@ int register_mem_sect_under_node(struct memory_block *mem_blk, int nid,
 
         if (!mem_blk)
                 return -EFAULT;
+
+        mem_blk->nid = nid;
         if (!node_online(nid))
                 return 0;