1 files changed, 3 insertions, 3 deletions
diff --git a/monkey/test.py b/monkey/test.py
index eb91e3c..4b0d192 100755
--- a/monkey/test.py
+++ b/monkey/test.py
@@ -8,7 +8,7 @@ import django
 from termcolor import colored
 
 from .action import parse
-from .edits import classify_edits, edit_graph, get_edits_from_traces
+from .edits import classify_edits, trace_graph, get_edits_from_traces
 from .graph import graphviz
 from .monkey import fix
 from prolog.engine import test
@@ -131,9 +131,9 @@ elif len(sys.argv) >= 3 and sys.argv[2] == 'info':
 elif len(sys.argv) == 4 and sys.argv[2] == 'graph':
     uid = int(sys.argv[3])
     user = User.objects.get(pk=uid)
-    actions = parse(Attempt.objects.get(problem=problem, user=user).trace)
+    attempt = Attempt.objects.get(problem=problem, user=user)
 
-    nodes, submissions, queries = edit_graph(actions)
+    nodes, submissions, queries = trace_graph(attempt.trace)
 
     def position(node):
         return (node.data[1]*150, node.data[0]*-60)