diff --git a/graph/graph.py b/graph/graph.py
index 03af12cbbd..da829c388f 100644
--- a/graph/graph.py
+++ b/graph/graph.py
@@ -120,7 +120,7 @@ def make_graph(pdus, room, filename_prefix):
 def get_pdus(host, room):
     transaction = json.loads(
         urllib2.urlopen(
-            "http://%s/context/%s/" % (host, room)
+            "http://%s/matrix/federation/v1/context/%s/" % (host, room)
         ).read()
     )