diff --git a/mm/memory_hotplug.c b/mm/memory_hotplug.c
index bb908289679e1a01e4cc35cdc78c0a50511aa93e..7a5fc89a865289b76af1df2e4c992a6dbf2ae2df 100644
--- a/mm/memory_hotplug.c
+++ b/mm/memory_hotplug.c
@@ -1458,7 +1458,7 @@ int __ref add_memory_resource(int nid, struct resource *res, mhp_t mhp_flags)
 	/* create memory block devices after memory was added */
 	ret = create_memory_block_devices(start, size, params.altmap, group);
 	if (ret) {
-		arch_remove_memory(start, size, NULL);
+		arch_remove_memory(start, size, params.altmap);
 		goto error_free;
 	}