diff --git a/kernel/ethosu_cancel_inference.h b/kernel/ethosu_cancel_inference.h
index 044692c..a21b9c0 100644
--- a/kernel/ethosu_cancel_inference.h
+++ b/kernel/ethosu_cancel_inference.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2022 ARM Limited.
+ * Copyright 2022-2023 Arm Limited and/or its affiliates
  *
  * This program is free software and is provided to you under the terms of the
  * GNU General Public License version 2 as published by the Free Software
@@ -35,10 +35,10 @@
  * Types
  ****************************************************************************/
 
-struct ethosu_core_cancel_inference_rsp;
+struct ethosu_core_msg_cancel_inference_rsp;
 struct ethosu_device;
-struct ethosu_uapi_cancel_inference_status;
 struct ethosu_inference;
+struct ethosu_uapi_cancel_inference_status;
 
 struct ethosu_cancel_inference {
 	struct ethosu_device                       *edev;
@@ -65,6 +65,7 @@
  * ethosu_cancel_inference_rsp() - Handle cancel inference response
  */
 void ethosu_cancel_inference_rsp(struct ethosu_device *edev,
-				 struct ethosu_core_cancel_inference_rsp *rsp);
+				 int msg_id,
+				 struct ethosu_core_msg_cancel_inference_rsp *rsp);
 
 #endif /* ETHOSU_CANCEL_INFERENCE_H */