IntelliJ code style keep method name and parameter on the same line

I'm trying to create a code style in IntelliJ to keep formatting consistent. At the moment, I have a problem where IntelliJ reformats a line very strangely.

                                                        |
someCollectionMethod().stream().filter(Objects::nonNull).map(...)
                                                        |         

The right margin is where the | are. IntelliJ reformats this to

someCollectionMethod().stream().filter
        (Objects::nonNull).map(...)

I find this to be incredibly strange and would much rather it take the method name to the next line as well like

someCollectionMethod().stream()
        .filter(Objects::nonNull).map(...)

I want the margin to be enforced but not in such a strange way. I can't find any setting about this.