Why remove this check ?
If mapreduce.reduce.memory.mb is not found in the config, why silently return the mapper memory MB for that instead ? Do you find cases where the reducer memory was not set and you're getting exceptions ?