diff --git a/kernel/ethosu_inference.c b/kernel/ethosu_inference.c
index 57529f9..ac617c3 100644
--- a/kernel/ethosu_inference.c
+++ b/kernel/ethosu_inference.c
@@ -41,8 +41,8 @@
 static int ethosu_inference_release(struct inode *inode,
 				    struct file *file);
 
-static unsigned int ethosu_inference_poll(struct file *file,
-					  poll_table *wait);
+static __poll_t ethosu_inference_poll(struct file *file,
+				      poll_table *wait);
 
 static long ethosu_inference_ioctl(struct file *file,
 				   unsigned int cmd,
@@ -184,16 +184,16 @@
 	return 0;
 }
 
-static unsigned int ethosu_inference_poll(struct file *file,
-					  poll_table *wait)
+static __poll_t ethosu_inference_poll(struct file *file,
+				      poll_table *wait)
 {
 	struct ethosu_inference *inf = file->private_data;
-	int ret = 0;
+	__poll_t ret = 0;
 
 	poll_wait(file, &inf->waitq, wait);
 
 	if (inf->done)
-		ret |= POLLIN;
+		ret |= EPOLLIN;
 
 	return ret;
 }