diff --git a/tests/addFastAlignedTM.py b/tests/addFastAlignedTM.py
index 4ec1155..213a8b2 100755
--- a/tests/addFastAlignedTM.py
+++ b/tests/addFastAlignedTM.py
@@ -78,7 +78,7 @@ examples = []
 start = time.time()
 with open(sourceFile) as source_file, open(lemmatizedSourceFile) as lemmatized_source_file, open(targetFile) as target_file, open(alignmentsFile) as alignments_file, open(sourceIdsFile) as source_ids_file:
     addedCount = 0
-    for lineNumber in range(totalExamples):
+    for lineNumber in range(sourceFileLength):
         if lineNumber % LEAVE_OUT == 0:
             sourceSentence = source_file.readline().strip()
             lemmatizedSourceSentence = lemmatized_source_file.readline().strip()