diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc
index 9763e624ce029..fb99fcfe5d3b4 100644
--- a/src/relay/op/tensor/transform.cc
+++ b/src/relay/op/tensor/transform.cc
@@ -3394,7 +3394,7 @@ Let B denote batch_dims, and data, indices shape be (X_0, X_1, ..., X_{N-1}),
 and it must be the case that X_0, ... X_{B-1} == Y_0, ... Y_{B-1}.
 
 The output will have shape
-(Y_0, ..., Y_{K-1}, X_{M+B}, ..., X_{N-1}), where M + B <= N. If M + B == N,
+(Y_0, ..., Y_{B-1}, ..., Y_{K-1}, X_{M+B}, ..., X_{N-1}), where M + B <= N. If M + B == N,
 output shape will simply be (Y_0, ..., Y_{K-1}).
 )code" TVM_ADD_FILELINE)
     .set_num_inputs(2)