DisableAutoGeneratedUIDs fails on keyBy

 @Test
public void testFailedUID() throws Exception {
    StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
    env.getConfig().disableAutoGeneratedUIDs();

    DataStream<String> data = env.fromCollection(Arrays.asList("1", "2", "3")).uid("source-uid");
    data.keyBy(i -> i)
            .map(i -> i).uid("map-uid");

    env.execute();
}

testFailedUID(twitch.creatoranalytics.sessions.StreamingJobTest)  Time elapsed: 0.008 sec  <<< ERROR!
java.lang.IllegalStateException: Auto generated UIDs have been disabled but no UID or hash has been assigned to operator Partition