diff --git a/core/ast/src/main/java/org/overture/ast/util/PTypeSet.java b/core/ast/src/main/java/org/overture/ast/util/PTypeSet.java
index <HASH>..<HASH> 100644
--- a/core/ast/src/main/java/org/overture/ast/util/PTypeSet.java
+++ b/core/ast/src/main/java/org/overture/ast/util/PTypeSet.java
@@ -39,9 +39,10 @@ public class PTypeSet extends TreeSet<PType>
 		add(t2);
 	}
 
-	public PTypeSet(List<PType> types)
+	public PTypeSet(List<PType> types, IAstAssistantFactory af)
 	{
 		super(new PTypeComparator());
+		assistantFactory = af;
 		addAll(types);
 	}
Add missing assistant from PTypeSet constructor.