diff --git a/drivers/platform/msm/gsi/gsi.c b/drivers/platform/msm/gsi/gsi.c
index 43757d5..3084f76 100644
--- a/drivers/platform/msm/gsi/gsi.c
+++ b/drivers/platform/msm/gsi/gsi.c
@@ -2916,9 +2916,9 @@
 	}
 	memset(ctx, 0, sizeof(*ctx));
 
-	/* For IPA offloaded WDI channels not required user_data pointer */
-	if (props->prot != GSI_CHAN_PROT_WDI2 &&
-		props->prot != GSI_CHAN_PROT_WDI3)
+	/* For IPA offloaded WDI/RTK/XDCI channels not required user_data pointer */
+	if (props->prot == GSI_CHAN_PROT_GPI ||
+		props->prot == GSI_CHAN_PROT_GCI)
 		user_data_size = props->ring_len / props->re_size;
 	else
 		user_data_size = props->re_size;