diff --git a/include/uapi/linux/nvgpu.h b/include/uapi/linux/nvgpu.h
index f3c750450..b3302e6f0 100644
--- a/include/uapi/linux/nvgpu.h
+++ b/include/uapi/linux/nvgpu.h
@@ -1231,7 +1231,7 @@ struct nvgpu_preemption_mode_args {
 
 #define NVGPU_IOCTL_CHANNEL_LAST	\
 	_IOC_NR(NVGPU_IOCTL_CHANNEL_SET_PREEMPTION_MODE)
-#define NVGPU_IOCTL_CHANNEL_MAX_ARG_SIZE sizeof(struct nvgpu_submit_gpfifo_args)
+#define NVGPU_IOCTL_CHANNEL_MAX_ARG_SIZE sizeof(struct nvgpu_alloc_gpfifo_ex_args)
 
 /*
  * /dev/nvhost-as-gpu device