diff --git a/tests/test_parse.py b/tests/test_parse.py
index d89a185..7d38b50 100644
--- a/tests/test_parse.py
+++ b/tests/test_parse.py
@@ -245,14 +245,13 @@
         new_code = token.code + space
         lines = new_code.split("\n")
         if len(lines) > 1:
-            curr_line += (len(lines) - 1)
+            curr_line += len(lines) - 1
             curr_column = len(lines[-1]) + 1
         else:
             curr_column += len(new_code)
         code += new_code
-    all_lines = code.split("\n")
     eof = SampleToken(None, "EOF", None)
-    eof.location = parse.ParseLocation(len(all_lines), len(all_lines[-1]), "")
+    eof.location = parse.ParseLocation(curr_line, curr_column - 1, "")
     return SampleSoup(all_tokens + [eof], code)