For setting the optical gain of an optical amplifier such as a Raman amplifier that is connected in a wavelength division multiplexing (WDM) system, the gain of the amplifier is made dependent on the states of optical polarizers connected to individual inputs of a WDM multiplexer. The polarizers can be actively controlled by a device connected to sense the output power of the Raman fiber at different wavelengths. For an appropriate control the optical gain can be given any desired shape such as for example a reasonable flatness. The control of the polarization states of the WDM-channels allows for the use of a single wavelength pump source of the amplifier, instead of the conventionally used multiwavelength source.