tiny refactor

2 files changed, 5 insertions, 5 deletions

diff --git a/pkg/gui/gui_test.go b/pkg/gui/gui_test.go
index c6f945ae9..677b77f7c 100644
--- a/pkg/gui/gui_test.go
+++ b/pkg/gui/gui_test.go
@@ -57,7 +57,8 @@ func Test(t *testing.T) {
 		test := test
 
 		t.Run(test.Name, func(t *testing.T) {
-			speeds := integration.GetTestSpeeds(test.Speed, updateSnapshots)
+			speedEnv := os.Getenv("SPEED")
+			speeds := integration.GetTestSpeeds(test.Speed, updateSnapshots, speedEnv)
 			testPath := filepath.Join(testDir, test.Name)
 			actualDir := filepath.Join(testPath, "actual")
 			expectedDir := filepath.Join(testPath, "expected")
diff --git a/pkg/integration/integration.go b/pkg/integration/integration.go
index e7e21bbd7..e433f1954 100644
--- a/pkg/integration/integration.go
+++ b/pkg/integration/integration.go
@@ -79,15 +79,14 @@ func TempLazygitPath() string {
 	return filepath.Join("/tmp", "lazygit", "test_lazygit")
 }
 
-func GetTestSpeeds(testStartSpeed float64, updateSnapshots bool) []float64 {
+func GetTestSpeeds(testStartSpeed float64, updateSnapshots bool, speedStr string) []float64 {
 	if updateSnapshots {
 		// have to go at original speed if updating snapshots in case we go to fast and create a junk snapshot
 		return []float64{1.0}
 	}
 
-	speedEnv := os.Getenv("SPEED")
-	if speedEnv != "" {
-		speed, err := strconv.ParseFloat(speedEnv, 64)
+	if speedStr != "" {
+		speed, err := strconv.ParseFloat(speedStr, 64)
 		if err != nil {
 			panic(err)
 		}