lib/provider/ec2.rb in aws_security_viz-0.1.5.pre.alpha.pre.87 vs lib/provider/ec2.rb in aws_security_viz-0.1.5.pre.alpha.pre.89

@@ -9,9 +9,13 @@
     }
 
     conn_opts[:aws_access_key_id] = options[:access_key]
     conn_opts[:aws_secret_access_key] = options[:secret_key]
 
+    if options[:session_token]
+      conn_opts[:aws_session_token] = options[:session_token]
+    end
+
     @compute = Fog::Compute::AWS.new conn_opts
   end
 
   def security_groups
     @compute.security_groups.reject { |sg|