diff --git a/include/trace/hooks/psci.h b/include/trace/hooks/psci.h
index a00925df75da..97d0fb1469a5 100644
--- a/include/trace/hooks/psci.h
+++ b/include/trace/hooks/psci.h
@@ -10,13 +10,13 @@
 
 #include <trace/hooks/vendor_hooks.h>
 
-DECLARE_HOOK(android_rvh_psci_tos_resident_on,
+DECLARE_RESTRICTED_HOOK(android_rvh_psci_tos_resident_on,
 	TP_PROTO(int cpu, bool *resident),
-	TP_ARGS(cpu, resident));
+	TP_ARGS(cpu, resident), 1);
 
-DECLARE_HOOK(android_rvh_psci_cpu_suspend,
+DECLARE_RESTRICTED_HOOK(android_rvh_psci_cpu_suspend,
 	TP_PROTO(u32 state, bool *deny),
-	TP_ARGS(state, deny));
+	TP_ARGS(state, deny), 1);
 
 /* macro versions of hooks are no longer required */