A multi-functional diffractive optical element (DOE) for redirecting light into a waveguide and providing higher order aberration correction is described. The multi-functional DOE may be positioned on, connected to, adjacent to, or within a waveguide, and in some examples is positioned at, or near, the exit pupil of the projector lens. In an example, a head-mounted display (HMD) is configured to output artificial reality content, comprising a waveguide configured to receive input light and configured to output the received input light to an eyebox. The HMD further comprises a projector configured to input light into the waveguide, the projector comprising a display, a projection lens, and a multi-functional diffractive optical element (DOE) configured to redirect light from the projector into the waveguide and provide higher order aberration correction of the light from the display.