I think you need to set the parallelism here. Have a look at how the operators that inject key extracting mappers do it.