diff --git a/drivers/of/fdt.c b/drivers/of/fdt.c
index e31816d49c7a..4bbdcf76df9f 100644
--- a/drivers/of/fdt.c
+++ b/drivers/of/fdt.c
@@ -333,6 +333,9 @@ static void __unflatten_device_tree(void *blob,
 
 	/* Allocate memory for the expanded device tree */
 	mem = dt_alloc(size + 4, __alignof__(struct device_node));
+	if (!mem)
+		return;
+
 	memset(mem, 0, size);
 
 	*(__be32 *)(mem + size) = cpu_to_be32(0xdeadbeef);