A system and method are disclosed for providing uniform color distribution of light emitted from a light source to an eye box in a near eye display (NED). An example of the system and method uses an optical element including two or more waveguides optimized to different colors of the visible light spectrum. The optical element further includes one or more polarization state generators for controlling the polarization of light incident on the waveguides to facilitate coupling of light into a matched waveguide, and to impede coupling of light into unmatched waveguides.