diff --git a/recbole/model/general_recommender/simplex.py b/recbole/model/general_recommender/simplex.py
index 9fa9e12c3..53a7929f7 100644
--- a/recbole/model/general_recommender/simplex.py
+++ b/recbole/model/general_recommender/simplex.py
@@ -74,7 +74,7 @@ def __init__(self, config, dataset):
             if self.aggregator == "self_attention":
                 self.W_q = nn.Linear(self.embedding_size, 1, bias=False)
         # dropout
-        self.dropout_prob = nn.Dropout(config["dropout_prob"])
+        self.dropout = nn.Dropout(config["dropout_prob"])
         self.require_pow = config["require_pow"]
         # l2 regularization loss
         self.reg_loss = EmbLoss()