1 files changed, 2 insertions, 8 deletions
diff --git a/Lib/trace.py b/Lib/trace.py
index 7ec0cf6bc6..12e1b49207 100644
--- a/Lib/trace.py
+++ b/Lib/trace.py
@@ -58,7 +58,7 @@ import token
 import tokenize
 import inspect
 import gc
-
+import dis
 try:
     import cPickle
     pickle = cPickle
@@ -379,13 +379,7 @@ def find_lines_from_code(code, strs):
     """Return dict where keys are lines in the line number table."""
     linenos = {}
 
-    line_increments = [ord(c) for c in code.co_lnotab[1::2]]
-    table_length = len(line_increments)
-    docstring = False
-
-    lineno = code.co_firstlineno
-    for li in line_increments:
-        lineno += li
+    for _, lineno in dis.findlinestarts(code):
         if lineno not in strs:
             linenos[lineno] = 1