diff --git a/.gitignore b/.gitignore
index 8d64d5d..b60dca9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,5 +1,6 @@
 .ipynb_checkpoints
 .DS_Store
+.idea
 
 __pycache__/
 
diff --git a/DeepPurpose/models.py b/DeepPurpose/models.py
index 93278cd..fa239a8 100644
--- a/DeepPurpose/models.py
+++ b/DeepPurpose/models.py
@@ -137,7 +137,7 @@ def __init__(self, encoding, **config):
 				raise AttributeError('Please use LSTM or GRU.')
 			direction = 2 if config['rnn_drug_bidirectional'] else 1
 			self.rnn = self.rnn.double()
-			self.fc1 = nn.Linear(config['rnn_drug_hid_dim'] * direction * n_size_p, config['hidden_dim_drug'])
+			self.fc1 = nn.Linear(config['rnn_drug_hid_dim'] * direction * n_size_d, config['hidden_dim_drug'])
 
 		if encoding == 'protein':
 			in_ch = [26] + config['cnn_target_filters']