Fix the wavelength sample issue