Option to keep adjusted white balance while doing before and after

Hi, sometimes I forget to set the white balance. When doing the before and after, the difference in WB makes the experience less useful

Would anyone else like to have the ability to keep the adjusted WB while using the before and after feature?

Thanks