diff --git a/rsa2/grader.py b/rsa2/grader.py
index b5e0f6f..4211957 100644
--- a/rsa2/grader.py
+++ b/rsa2/grader.py
@@ -42,7 +42,7 @@ def generate(random):
     })
 
 def grade(random, key):
-    n, salt = get_problem(random)
+    p, q, e, salt = get_problem(random)
     if key.find("%s_%s" % (flag, salt)) >= 0:
         return True, "Correct!"
     return False, "Nope."