diff --git a/pypreprocessor/__init__.py b/pypreprocessor/__init__.py
index b0b2a07..ff95e3f 100644
--- a/pypreprocessor/__init__.py
+++ b/pypreprocessor/__init__.py
@@ -237,11 +237,7 @@ def rewrite_traceback(self):
         trace = traceback.format_exc().splitlines()
         index = 0
         for line in trace:
-            if index == (len(trace) - 2):
-                print(line.replace("<string>", self.input))
-            else:
-                print(line)
-            index += 1
+            print(line.replace("\"<string>\"", self.input))
 
     # parsing/processing
     def parse(self):