diff --git a/rsa1/grader.py b/rsa1/grader.py
index 7bc9d86..b864bbf 100644
--- a/rsa1/grader.py
+++ b/rsa1/grader.py
@@ -43,7 +43,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."