diff --git a/model/classification/textvdcnn.py b/model/classification/textvdcnn.py
index a34d02e..302c4b3 100644
--- a/model/classification/textvdcnn.py
+++ b/model/classification/textvdcnn.py
@@ -78,7 +78,7 @@ def get_parameter_optimizer_dict(self):
             params.append({'params': self.batch_norms[i].parameters()})
         params.append({'params': self.linear1.parameters()})
         params.append({'params': self.linear2.parameters()})
-        params.append({'params': self.linear3.parameters()})
+        params.append({'params': self.linear.parameters()})
         return params
 
     def update_lr(self, optimizer, epoch):