Fix 2D vs. 1D wavelength bug