diff --git a/date_selection.py b/date_selection.py
index f26ea7ec8431eda886e1eed355186aa78ad217aa..d180e51936fbf3500e5bfc2f9837dfdbabfef20e 100644
--- a/date_selection.py
+++ b/date_selection.py
@@ -1,5 +1,6 @@
 from collections import Counter
 from datetime import datetime
+import random
 
 import igraph
 
@@ -34,13 +35,13 @@ def rank_dates_by_wilson(articles, start_date, end_date):
         edges.append((pub_date, mentioned_date, edge_weight))
 
     # create a graph from the edge list
-    g = igraph.Graph.TupleList(edges, directed=True)
+    g = igraph.Graph.TupleList(edges, directed=True, edge_attrs='weight')
     vertex_names = g.vs['name']
 
     # igraph.plot(g, layout='kk', vertex_label=g.vs['name'], bbox=(3000, 3000))
 
     # rank vertices by pagerank score
-    pagerank_scores = g.pagerank()
+    pagerank_scores = g.pagerank(directed=True, weights=g.es['weight'])
     ranked_dates = util.rank(vertex_names, scores=pagerank_scores)
 
     return ranked_dates