From a7d2dc0ac94c87e572e276bd5722d232080568d3 Mon Sep 17 00:00:00 2001
From: Kunal Gupta <39487888+iamKunalGupta@users.noreply.github.com>
Date: Thu, 21 Mar 2024 02:16:50 +0530
Subject: [PATCH] chore(snowflake): remove conditional token

---
 flow/connectors/snowflake/qrep.go | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/flow/connectors/snowflake/qrep.go b/flow/connectors/snowflake/qrep.go
index d8545c94a7..43445ffb83 100644
--- a/flow/connectors/snowflake/qrep.go
+++ b/flow/connectors/snowflake/qrep.go
@@ -140,13 +140,8 @@ func (c *SnowflakeConnector) createExternalStage(ctx context.Context, stageName
 		return "", err
 	}
 	if s3Int == "" {
-		awsTokenParam := ""
-		if creds.AWS.SessionToken != "" {
-			awsTokenParam = fmt.Sprintf(" AWS_TOKEN='%s'", creds.AWS.SessionToken)
-		}
-		credsStr := fmt.Sprintf("CREDENTIALS=(AWS_KEY_ID='%s' AWS_SECRET_KEY='%s'%s)",
-			creds.AWS.AccessKeyID, creds.AWS.SecretAccessKey, awsTokenParam)
-
+		credsStr := fmt.Sprintf("CREDENTIALS=(AWS_KEY_ID='%s' AWS_SECRET_KEY='%s' AWS_TOKEN='%s')",
+			creds.AWS.AccessKeyID, creds.AWS.SecretAccessKey, creds.AWS.SessionToken)
 		stageStatement := `
 		CREATE OR REPLACE STAGE %s
 		URL = '%s'