Make map_reduce @spec stricter