diff --git a/pkg/integration/tests/commit/discard_old_file_changes.go b/pkg/integration/tests/commit/discard_old_file_changes.go
index 2dc8006bb..d7f4acdc1 100644
--- a/pkg/integration/tests/commit/discard_old_file_changes.go
+++ b/pkg/integration/tests/commit/discard_old_file_changes.go
@@ -131,9 +131,9 @@ var DiscardOldFileChanges = NewIntegrationTest(NewIntegrationTestArgs{
 				Equals("    A file2ToRemove"),
 			).
 			NavigateToLine(Contains("multiLineFile")).
-			PressEnter()
+			Press(keys.Universal.FocusMainView)
 
-		t.Views().PatchBuilding().
+		t.Views().Main().
 			IsFocused().
 			SelectedLine(
 				Contains("+this file has"),
diff --git a/pkg/integration/tests/demo/custom_patch.go b/pkg/integration/tests/demo/custom_patch.go
index 24e05c633..34516f73b 100644
--- a/pkg/integration/tests/demo/custom_patch.go
+++ b/pkg/integration/tests/demo/custom_patch.go
@@ -52,9 +52,9 @@ var CustomPatch = NewIntegrationTest(NewIntegrationTestArgs{
 					IsFocused().
 					NavigateToLine(Contains("users.go")).
 					Wait(1000).
-					PressEnter().
+					Press(keys.Universal.FocusMainView).
 					Tap(func() {
-						t.Views().PatchBuilding().
+						t.Views().Main().
 							IsFocused().
 							NavigateToLine(Contains("TODO")).
 							Wait(500).
diff --git a/pkg/integration/tests/demo/stage_lines.go b/pkg/integration/tests/demo/stage_lines.go
index 1304d39e5..4881722a7 100644
--- a/pkg/integration/tests/demo/stage_lines.go
+++ b/pkg/integration/tests/demo/stage_lines.go
@@ -56,9 +56,9 @@ var StageLines = NewIntegrationTest(NewIntegrationTestArgs{
 
 		t.Views().Files().
 			IsFocused().
-			PressEnter()
+			Press(keys.Universal.FocusMainView)
 
-		t.Views().Staging().
+		t.Views().Main().
 			IsFocused().
 			Press(keys.Universal.ToggleRangeSelect).
 			PressFast(keys.Universal.NextItem).