diff --git a/drivers/of/pdt.c b/drivers/of/pdt.c
index bc5b3990f6ed114f50103c8a8b060dfc69cc6aed..07cc1d678e4d3da7d810c330c8bb4053411bd885 100644
--- a/drivers/of/pdt.c
+++ b/drivers/of/pdt.c
@@ -229,7 +229,7 @@ static struct device_node * __init of_pdt_build_tree(struct device_node *parent,
 	return ret;
 }
 
-static void *kernel_tree_alloc(u64 size, u64 align)
+static void * __init kernel_tree_alloc(u64 size, u64 align)
 {
 	return prom_early_alloc(size);
 }