Minor bugfix to whitening.