diff --git a/deepchem/models/graph_models.py b/deepchem/models/graph_models.py
index cf72262b5f..dc13f7ed86 100644
--- a/deepchem/models/graph_models.py
+++ b/deepchem/models/graph_models.py
@@ -635,7 +635,7 @@ def call(self, inputs, training=False):
     if self.batch_norms[-1] is not None:
       dense = self.batch_norms[-1](dense, training=training)
     if training and self.dropouts[-1] is not None:
-      dense = self.dropouts[1](dense, training=training)
+      dense = self.dropouts[-1](dense, training=training)
     neural_fingerprint = self.graph_gather([dense, degree_slice, membership] +
                                            deg_adjs)
     if self.mode == 'classification':