A method to improve codebook performance for non-linear arrays is described. The method includes determining a unitary matrix for a plurality of transmission antennas arranged in a given array type, the unitary matrix being determined based on a codebook, where the given array type is configured to steer beams in at least one of elevation and azimuth. The method also includes applying the determined unitary matrix to a signal to be transmitted across the plurality of transmission antennas. Apparatus and computer readable media are also described.