diff --git a/lib/capybara-screenshot/rspec.rb b/lib/capybara-screenshot/rspec.rb
index 559fe8a..99750fe 100644
--- a/lib/capybara-screenshot/rspec.rb
+++ b/lib/capybara-screenshot/rspec.rb
@@ -90,7 +90,7 @@ def failed?(example)
     Capybara::Screenshot.final_session_name = nil
   end
 
-  config.after do |example_from_block_arg|
+  config.after(type: :feature) do |example_from_block_arg|
     # RSpec 3 no longer defines `example`, but passes the example as block argument instead
     example = config.respond_to?(:expose_current_running_example_as) ? example_from_block_arg : self.example