1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/base/memory.c b/drivers/base/memory.c
index 1c617623c8ae..9e59f6535c44 100644
--- a/drivers/base/memory.c
+++ b/drivers/base/memory.c
@@ -321,7 +321,9 @@ store_mem_state(struct device *dev,
         struct memory_block *mem = to_memory_block(dev);
         int ret, online_type;
 
-        lock_device_hotplug();
+        ret = lock_device_hotplug_sysfs();
+        if (ret)
+                return ret;
 
         if (!strncmp(buf, "online_kernel", min_t(int, count, 13)))
                 online_type = ONLINE_KERNEL;