diff --git a/pkg/gui/controllers/toggle_whitespace_action.go b/pkg/gui/controllers/toggle_whitespace_action.go
index a1ac0c8da..67bb59d86 100644
--- a/pkg/gui/controllers/toggle_whitespace_action.go
+++ b/pkg/gui/controllers/toggle_whitespace_action.go
@@ -27,6 +27,6 @@ func (self *ToggleWhitespaceAction) Call() error {
 
 	self.c.UserConfig().Git.IgnoreWhitespaceInDiffView = !self.c.UserConfig().Git.IgnoreWhitespaceInDiffView
 
-	self.c.Context().CurrentSide().HandleFocus(types.OnFocusOpts{})
+	self.c.Context().CurrentSide().HandleRenderToMain()
 	return nil
 }
diff --git a/pkg/integration/tests/ui/toggle_whitespace_keeps_unfocused_selection_dimmed.go b/pkg/integration/tests/ui/toggle_whitespace_keeps_unfocused_selection_dimmed.go
index 30223c3b8..9ffbd2878 100644
--- a/pkg/integration/tests/ui/toggle_whitespace_keeps_unfocused_selection_dimmed.go
+++ b/pkg/integration/tests/ui/toggle_whitespace_keeps_unfocused_selection_dimmed.go
@@ -26,9 +26,6 @@ var ToggleWhitespaceKeepsUnfocusedSelectionDimmed = NewIntegrationTest(NewIntegr
 			Press(keys.Universal.ToggleWhitespaceInDiffView)
 
 		t.Views().Files().
-			/* EXPECTED:
 			SelectionIsInactive()
-			ACTUAL: */
-			SelectionIsActive()
 	},
 })