--- board/raspberrypi/rpi/rpi.c.orig	2016-10-28 12:12:02.771494000 -0700
+++ board/raspberrypi/rpi/rpi.c	2016-10-28 12:14:46.730250000 -0700
@@ -21,6 +21,7 @@
 #include <dm/platform_data/serial_bcm283x_mu.h>
 #ifdef CONFIG_ARM64
 #include <asm/armv8/mmu.h>
+#include <asm/psci.h>
 #endif
 
 DECLARE_GLOBAL_DATA_PTR;
@@ -507,6 +508,9 @@
 
 int ft_board_setup(void *blob, bd_t *bd)
 {
+#if defined(CONFIG_ARMV8_PSCI)
+	int ret;
+#endif
 	/*
 	 * For now, we simply always add the simplefb DT node. Later, we
 	 * should be more intelligent, and e.g. only do this if no enabled DT
@@ -514,5 +518,10 @@
 	 */
 	lcd_dt_simplefb_add_node(blob);
 
+#if defined(CONFIG_ARMV8_PSCI)
+    	ret = psci_update_dt(blob);
+	if (ret)
+		return ret;
+#endif
 	return 0;
 }