diff --git a/drivers/misc/optee_smc.c b/drivers/misc/optee_smc.c
index 3d99ccc6646..43a33aef964 100644
--- a/drivers/misc/optee_smc.c
+++ b/drivers/misc/optee_smc.c
@@ -29,6 +29,7 @@
 #include <errno.h>
 #include <syslog.h>
 #include <string.h>
+#include <sched.h>
 
 #include "optee.h"
 #include "optee_smc.h"