lib/aws_su.rb in aws_su-0.1.1 vs lib/aws_su.rb in aws_su-0.1.2

@@ -59,9 +59,11 @@
     @profile = options[:profile]
     @duration = options[:duration].nil? ? DURATION : options[:duration]
     @token_ttl = calculate_session_expiry(@duration)
 
     region = AWSConfig.profiles.first[1][:region]
+    region = AWSConfig.profiles[@profile][:region] unless
+        AWSConfig.profiles[@profile][:region].nil?
     @region = options[:region].nil? ? region : options[:region]
     raise('Unable to determine region') if @region.nil?
 
     export_aws_sudo_file
     assume_role