diff --git a/src/de/dhbwstuttgart/typeinference/typeAlgo/TYPEStmt.java b/src/de/dhbwstuttgart/typeinference/typeAlgo/TYPEStmt.java
index 8d87218d..0cadc085 100644
--- a/src/de/dhbwstuttgart/typeinference/typeAlgo/TYPEStmt.java
+++ b/src/de/dhbwstuttgart/typeinference/typeAlgo/TYPEStmt.java
@@ -349,7 +349,7 @@ public class TYPEStmt implements StatementVisitor{
         */
         methodConstraint.add(new Pair(forMethod.receiver.getType(), assumption.getReceiverType(resolver),
                 PairOperator.SMALLERDOT));
-        methodConstraint.add(new Pair(assumption.getReturnType(), forMethod.getType(),
+        methodConstraint.add(new Pair(assumption.getReturnType(resolver), forMethod.getType(),
                 PairOperator.EQUALSDOT));
         methodConstraint.addAll(generateParameterConstraints(forMethod, assumption, info, resolver));
         return methodConstraint;
@@ -442,7 +442,7 @@ public class TYPEStmt implements StatementVisitor{
     protected Constraint<Pair> generateConstructorConstraint(NewClass forConstructor, MethodAssumption assumption,
                                                              TypeInferenceBlockInformation info, GenericsResolver resolver){
         Constraint methodConstraint = new Constraint();
-        methodConstraint.add(new Pair(assumption.getReturnType(), forConstructor.getType(),
+        methodConstraint.add(new Pair(assumption.getReturnType(resolver), forConstructor.getType(),
                 PairOperator.SMALLERDOT));
         methodConstraint.addAll(generateParameterConstraints(forConstructor, assumption, info, resolver));
         return methodConstraint;
diff --git a/test/javFiles/LambdaRunnable.jav b/test/javFiles/LambdaRunnable.jav
index 982680c3..37924ea2 100644
--- a/test/javFiles/LambdaRunnable.jav
+++ b/test/javFiles/LambdaRunnable.jav
@@ -1,3 +1,6 @@
+import java.lang.Runnable;
+import java.lang.String;
+import java.lang.System;
 
 public class LamRunnable{
 
diff --git a/test/visualisation/ToString.java b/test/visualisation/ToString.java
new file mode 100644
index 00000000..00b8ddea
--- /dev/null
+++ b/test/visualisation/ToString.java
@@ -0,0 +1,5 @@
+package visualisation;
+
+public class ToString {
+
+}