Is this behaviour of retry and groupBy expected in Reactor?

gistfile1.java

 var flux = Flux.range(0, 10);
        var errorDone = new AtomicBoolean(false);

        var count = flux.groupBy(i -> i % 2).flatMap(groupedFlux -> {
            if (groupedFlux.key() == 0) {
                return groupedFlux.flatMap(x -> {
                    if (x == 4 && !errorDone.get()) {
                        errorDone.set(true);
                        return Mono.error(() -> new RuntimeException());
                    } else {
                        return Mono.just(x);
                    }
                }).retry();
            } else {
                return groupedFlux;
            }
        }).count().block();
        System.out.println(count);

The issue with this is that I would expect the sub flow to recover (because of the retry). But it won't. This will only work if I put retry() before the count(). In the latter situation the code prints 14... as I would have expected by putting the retry as you can see in the snippet...