diff --git a/pkg/git/init.go b/pkg/git/init.go
index 30ffa1ae..bf38b55e 100644
--- a/pkg/git/init.go
+++ b/pkg/git/init.go
@@ -48,7 +48,7 @@ func Init(repoPath string, commit bool, gitName, gitEmail string) error {
 	return nil
 }
 
-// Set the default branch for the passed repo as "main"
+// SwitchToMain sets the default branch to "main".
 func SwitchToMain(repo *git.Repository) error {
 	ref := plumbing.NewSymbolicReference(plumbing.HEAD, plumbing.ReferenceName("refs/heads/main"))
 	if err := repo.Storer.SetReference(ref); err != nil {
diff --git a/pkg/git/init_test.go b/pkg/git/init_test.go
index 29fcb057..f6f44ad3 100644
--- a/pkg/git/init_test.go
+++ b/pkg/git/init_test.go
@@ -8,17 +8,15 @@ import (
 	"github.com/go-git/go-git/v5/storage/memory"
 )
 
-// Verify that SwitchToMain has the desired effect on the repo
 func TestSwitchToMain(t *testing.T) {
-	storage := memory.NewStorage()
-	repo, err := git.Init(storage, nil)
+	repo, err := git.Init(memory.NewStorage(), nil)
 	if err != nil {
 		t.Fatalf("failed to create in-memory repository: %v", err)
 	}
-	err = SwitchToMain(repo)
-	if err != nil {
+	if err = SwitchToMain(repo); err != nil {
 		t.Fatalf("SwitchToMain failed: %v", err)
 	}
+
 	ref, err := repo.Reference(plumbing.HEAD, false)
 	if err != nil {
 		t.Fatalf("failed to get HEAD reference: %v", err)
@@ -26,6 +24,7 @@ func TestSwitchToMain(t *testing.T) {
 	if ref.Target().String() != "refs/heads/main" {
 		t.Errorf("expected HEAD to point to 'refs/heads/main', got %s", ref.Target().String())
 	}
+
 	cfg, err := repo.Config()
 	if err != nil {
 		t.Fatalf("failed to get repository config: %v", err)