Move monkey/monkey.py to monkey/__init__.py

2 files changed, 2 insertions, 2 deletions

diff --git a/monkey/monkey.py b/monkey/__init__.py
index 6577450..6577450 100755
--- a/monkey/monkey.py
+++ b/monkey/__init__.py
diff --git a/monkey/test.py b/monkey/test.py
index 6736702..9936cdc 100755
--- a/monkey/test.py
+++ b/monkey/test.py
@@ -9,7 +9,7 @@ from termcolor import colored
 from db.models import CodeqUser, Problem, Solution
 from .edits import classify_edits, trace_graph
 from .graph import graphviz
-from .monkey import fix, fix_hints
+from . import fix, fix_hints
 from prolog.util import annotate, compose, stringify
 import server.problems
 from .util import indent
@@ -20,13 +20,13 @@ if len(sys.argv) < 2:
     sys.exit(1)
 pid = int(sys.argv[1])
 
-name = Problem.get(id=pid).name
 language, problem_group, problem = Problem.get_identifier(pid)
 
 solved_problems = [p for p in CodeqUser.solved_problems(1, language)
                            if p != (problem_group, problem)]
 other_solutions = server.problems.solutions_for_problems(language, solved_problems)
 problem_module = server.problems.load_problem(language, problem_group, problem, 'common')
+name = server.problems.load_problem(language, problem_group, problem, 'en').name
 
 # Testing function.
 def test(code):