2 files changed, 3 insertions, 3 deletions

diff --git a/arch/ia64/kernel/uncached.c b/arch/ia64/kernel/uncached.c
index c58e933694d5..a7be4f203420 100644
--- a/arch/ia64/kernel/uncached.c
+++ b/arch/ia64/kernel/uncached.c
@@ -196,7 +196,7 @@ unsigned long uncached_alloc_page(int starting_nid)
         nid = starting_nid;
 
         do {
-                if (!node_online(nid))
+                if (!node_state(nid, N_HIGH_MEMORY))
                         continue;
                 uc_pool = &uncached_pools[nid];
                 if (uc_pool->pool == NULL)
@@ -268,7 +268,7 @@ static int __init uncached_init(void)
 {
         int nid;
 
-        for_each_online_node(nid) {
+        for_each_node_state(nid, N_ONLINE) {
                 uncached_pools[nid].pool = gen_pool_create(PAGE_SHIFT, nid);
                 mutex_init(&uncached_pools[nid].add_chunk_mutex);
         }
diff --git a/drivers/char/mspec.c b/drivers/char/mspec.c
index 04ac155d3a07..82f2e27dca7d 100644
--- a/drivers/char/mspec.c
+++ b/drivers/char/mspec.c
@@ -362,7 +362,7 @@ mspec_init(void)
                 is_sn2 = 1;
                 if (is_shub2()) {
                         ret = -ENOMEM;
-                        for_each_online_node(nid) {
+                        for_each_node_state(nid, N_ONLINE) {
                                 int actual_nid;
                                 int nasid;
                                 unsigned long phys;