diff --git a/java/typeMigration/testData/inspections/guava/convertImmutableCollections_after.java b/java/typeMigration/testData/inspections/guava/convertImmutableCollections_after.java
index 88582d93a397..d6e65b28e74e 100644
--- a/java/typeMigration/testData/inspections/guava/convertImmutableCollections_after.java
+++ b/java/typeMigration/testData/inspections/guava/convertImmutableCollections_after.java
@@ -7,7 +7,7 @@ class CollectList {
   public <T> List<T> toList(Stream<T> p) { return p.collect(Collectors.toList()); }
   public <T> List<T> toSortedList(Stream<T> p, Comparator<T> c) { return p.sorted(c).collect(Collectors.toList()); }
   public <T> Set<T> toSet(Stream<T> p) { return p.collect(Collectors.toSet()); }
-  public <T> Set<T> toSortedSet(Stream<T> p, Comparator<T> c) { return p.collect(Collectors.toCollection(TreeSet::new)); }
+  public <T> Set<T> toSortedSet(Stream<T> p, Comparator<T> c) { return p.collect(Collectors.toCollection(() -> new TreeSet<>(c))); }
   public <T> Map<T, T> toMap(Stream<T> p) { return p.collect(Collectors.toMap(Function.identity(), input -> input)); }
 
   void m(Stream<String> fluentIterable) {