From c86c1527f0c4d9efc25b7af0436e4cb2b79723b6 Mon Sep 17 00:00:00 2001
From: Max Cai <zhiyuan.cai@gmail.com>
Date: Tue, 2 Apr 2024 10:51:01 -0400
Subject: [PATCH] address comments

---
 internal/wrapper/wrapper.go | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/internal/wrapper/wrapper.go b/internal/wrapper/wrapper.go
index dd3b77d..dff0814 100644
--- a/internal/wrapper/wrapper.go
+++ b/internal/wrapper/wrapper.go
@@ -53,10 +53,8 @@ func RunTerraform(args []string, arch string) (int, error) {
 
 	log.Printf("version '%s' matches all constraints", matchingRelease.Version)
 
-	err = checkStateCommand(args, matchingRelease.Version)
-	if err != nil {
-		log.SetOutput(os.Stderr)
-		log.Fatal("error: ", err)
+	if err := checkStateCommand(args, matchingRelease.Version); err != nil {
+		return 1, err
 	}
 
 	executablePath, err := client.DownloadRelease(matchingRelease, runtime.GOOS, arch)