sorry for the inaccurate description, in the current implementation, if you add the checking in lookup, the issue in the second place (ShuffleMapTask) will not happen....