diff --git a/internal/exp/diffview/diffview_test.go b/internal/exp/diffview/diffview_test.go
index cd60838244a7504b6ca12cdd692dd7f4d7fe37df..af15e5f2821bc29ddc38d07f2d0b0c0a782166a9 100644
--- a/internal/exp/diffview/diffview_test.go
+++ b/internal/exp/diffview/diffview_test.go
@@ -70,14 +70,14 @@ var (
 	}
 	SmallWidthFunc = func(dv *diffview.DiffView) *diffview.DiffView {
 		return dv.
-			Before("text.txt", TestMultipleHunksBefore).
-			After("text.txt", TestMultipleHunksAfter).
+			Before("main.go", TestMultipleHunksBefore).
+			After("main.go", TestMultipleHunksAfter).
 			Width(40)
 	}
 	LargeWidthFunc = func(dv *diffview.DiffView) *diffview.DiffView {
 		return dv.
-			Before("text.txt", TestMultipleHunksBefore).
-			After("text.txt", TestMultipleHunksAfter).
+			Before("main.go", TestMultipleHunksBefore).
+			After("main.go", TestMultipleHunksAfter).
 			Width(120)
 	}