Trailing whitespaces not highlighted when partially staging

**Describe the bug**
I can see the trailing whitespaces marked red when I select a file for staging, but when I press `` on it to partially stage, the highlighting disappears.

**To Reproduce**
Steps to reproduce the behavior:
1. Have trailing whitespaces in a file
2. Go to `Files` tab and select the file
3. See the trailing whitespaces highlighted in red
4. Press ``
5. Highlighting disappears

**Expected behavior**
Highlighting should remain visible.

**Desktop (please complete the following information):**
- OS: Arch linux
- Lazygit Version v0.31.4.d5c1cfda