diff --git a/pkg/integration/tests/staging/stage_partial_block_of_changes_first_lines.go b/pkg/integration/tests/staging/stage_partial_block_of_changes_first_lines.go
index 56c285cec..0588184a0 100644
--- a/pkg/integration/tests/staging/stage_partial_block_of_changes_first_lines.go
+++ b/pkg/integration/tests/staging/stage_partial_block_of_changes_first_lines.go
@@ -50,8 +50,7 @@ var StagePartialBlockOfChangesFirstLines = NewIntegrationTest(NewIntegrationTest
 			PressPrimaryAction().
 			NavigateToLine(Contains("+2b")).
 			PressPrimaryAction().
-			// TODO: we expect +3b to be selected, but it isn't, so navigate to it
-			NavigateToLine(Contains("+3b")).
+			SelectedLines(Contains("+3b")).
 			PressPrimaryAction()
 
 		t.Views().StagingSecondary().