diff --git a/lib/solidus_dev_support/rspec/capybara.rb b/lib/solidus_dev_support/rspec/capybara.rb
index 5d3f0d66..b4bf2dbb 100644
--- a/lib/solidus_dev_support/rspec/capybara.rb
+++ b/lib/solidus_dev_support/rspec/capybara.rb
@@ -21,4 +21,10 @@
   end
 end
 
+require 'capybara-screenshot/rspec'
+
+Capybara::Screenshot.register_driver(:solidus_chrome_headless) do |driver, path|
+  driver.browser.save_screenshot(path)
+end
+
 require 'spree/testing_support/capybara_ext'
diff --git a/lib/solidus_dev_support/rspec/feature_helper.rb b/lib/solidus_dev_support/rspec/feature_helper.rb
index 7f1328af..c17ff943 100644
--- a/lib/solidus_dev_support/rspec/feature_helper.rb
+++ b/lib/solidus_dev_support/rspec/feature_helper.rb
@@ -8,8 +8,6 @@
 #
 
 require 'solidus_dev_support/rspec/rails_helper'
-
-require 'capybara-screenshot/rspec'
 require 'solidus_dev_support/rspec/capybara'
 
 def dev_support_assets_preload