Minor correction

Corrected np.choose