We need to do the same in the map directive.