lib/flamegraph.rb in flamegraph-0.0.9 vs lib/flamegraph.rb in flamegraph-0.1.0

@@ -1,17 +1,17 @@
 require "json"
 
 if RUBY_VERSION >= "2.1.0".freeze
   begin
     require "stackprof"
-  rescue
+  rescue LoadError
     STDERR.puts "Please require the stackprof gem falling back to fast_stack"
     require "fast_stack"
   end
 else
   begin
     require "fast_stack"
-  rescue
+  rescue LoadError
     STDERR.puts "Please require the fast_stack gem, note flamegraph is only supported on Ruby 2.0 and above"
   end
 end
 
 require "flamegraph/version"