diff --git a/osbenchmark/client.py b/osbenchmark/client.py
index 21e535fe8..f779a5999 100644
--- a/osbenchmark/client.py
+++ b/osbenchmark/client.py
@@ -263,7 +263,7 @@ def parse_aws_log_in_params(self):
             aws_log_in_dict["region"] = self.client_options.get("region")
             aws_log_in_dict["service"] = self.client_options.get("service")
             # optional: applicable only for role-based access
-            aws_log_in_dict["aws_session_token"] = os.environ.get("aws_session_token")
+            aws_log_in_dict["aws_session_token"] = self.client_options.get("aws_session_token")
         if (not aws_log_in_dict["aws_access_key_id"] or not aws_log_in_dict["aws_secret_access_key"]
                 or not aws_log_in_dict["service"] or not aws_log_in_dict["region"]):
             self.logger.error("Invalid amazon aws log in parameters, required input aws_access_key_id, "