diff --git a/arch/s390/mm/init.c b/arch/s390/mm/init.c
index 3fa3e532361227ad134f32b46c6c0db58d9d1240..34bd72d60fb9f0881e98b97ac2d5334f739f754d 100644
--- a/arch/s390/mm/init.c
+++ b/arch/s390/mm/init.c
@@ -229,6 +229,9 @@ int arch_add_memory(int nid, u64 start, u64 size, struct vmem_altmap *altmap,
 	unsigned long size_pages = PFN_DOWN(size);
 	int rc;
 
+	if (WARN_ON_ONCE(restrictions->altmap))
+		return -EINVAL;
+
 	rc = vmem_add_mapping(start, size);
 	if (rc)
 		return rc;