IntelliJ inference bug

IntelliJInferenceBug.java

import java.util.HashMap;
import java.util.Map;
import java.util.SortedMap;
import java.util.TreeMap;
import java.util.stream.Collectors;

public class IntelliJInferenceBug {

    public static void main(String[] args) {
        Map<String, Integer> map = new HashMap<>();
        map.put("one", 1);
        map.put("two", 2);
        map.put("three", 3);
        map.put("four", 4);

        System.out.println(map);

        SortedMap<String, Integer> sortedMap = map.entrySet().stream()
                                                  .collect(Collectors.toMap(e -> e.getKey(),
                                                                            Map.Entry::getValue,
                                                                            (u, v) -> u,
                                                                            TreeMap::new));

        System.out.println(sortedMap);
    }

}