diff --git a/src/main/java/de/dhbwstuttgart/inferWildcards/JavaTXCompilerWildcards.java b/src/main/java/de/dhbwstuttgart/inferWildcards/JavaTXCompilerWildcards.java
index b0902669..dd0f5481 100644
--- a/src/main/java/de/dhbwstuttgart/inferWildcards/JavaTXCompilerWildcards.java
+++ b/src/main/java/de/dhbwstuttgart/inferWildcards/JavaTXCompilerWildcards.java
@@ -100,9 +100,8 @@ public class JavaTXCompilerWildcards
 				throw new IOException("Could not create output directory at: " + outputDir.getAbsolutePath());
 		}
 		
-		// TODO for all result sets
 		List<ResultSet> typeInference = typeInference();
-		ResultSet resultSet = typeInference.get(0);
+		ResultSet resultSet = typeInference.get(0); // Use the first available result sets
 		
 		for (Map.Entry<File, SourceFile> e : sourceFiles.entrySet()) {
 			List<Map.Entry<? extends TypePlaceholder, ? extends RefTypeOrTPHOrWildcardOrGeneric>> list = tphMap.entrySet().stream().filter(d -> FindInAstVisitor.find(e.getValue(), d.getKey())).sorted( (c1, c2) -> Integer.compare(c1.getValue().getOffset().getStartIndex(),