diff --git a/graphsage/encoders.py b/graphsage/encoders.py
index e5d578a..9094b1c 100644
--- a/graphsage/encoders.py
+++ b/graphsage/encoders.py
@@ -35,7 +35,6 @@ class Encoder(nn.Module):
         Generates embeddings for a batch of nodes.
 
         nodes     -- list of nodes
-        mode      -- string desiginating the mode of the nodes
         """
         neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[node] for node in nodes], 
                 self.num_sample)