Fix bug in white balance