diff --git a/mindspore/nn/probability/distribution/categorical.py b/mindspore/nn/probability/distribution/categorical.py
index 0c41e717f1..970647b5dc 100644
--- a/mindspore/nn/probability/distribution/categorical.py
+++ b/mindspore/nn/probability/distribution/categorical.py
@@ -29,7 +29,7 @@ class Categorical(Distribution):
         probs (Tensor, list, numpy.ndarray, Parameter): Event probabilities.
         logits (Tensor, list, numpy.ndarray, Parameter, float): Event log-odds.
         seed (int): The global seed is used in sampling. Global seed is used if it is None. Default: None.
-        dtype (mstype.int32): The type of the distribution. Default: mstype.int32.
+        dtype (mindspore.dtype): The type of the distribution. Default: mstype.int32.
         name (str): The name of the distribution. Default: Categorical.
 
     Note: