diff --git a/implementations/Python/align_trace.py b/implementations/Python/align_trace.py
index 51dd0a5..65b8f3f 100644
--- a/implementations/Python/align_trace.py
+++ b/implementations/Python/align_trace.py
@@ -3,12 +3,11 @@ import sys
 SEP = '•'
 
 lines = sys.stdin.readlines()
-
-indicies = [line.index(SEP) for line in lines if SEP in line]
+indicies = [line.find(SEP) for line in lines]
 
 MAX = max(indicies)
 
-prefix_counts = [MAX - i for i in indicies]
+prefix_counts = (MAX - i for i in indicies)
 
 for count, line in zip(prefix_counts, lines):
     print(' ' * count, line, sep='', end = '')