diff --git a/lib/cucumber/rails/rspec.rb b/lib/cucumber/rails/rspec.rb
index <HASH>..<HASH> 100644
--- a/lib/cucumber/rails/rspec.rb
+++ b/lib/cucumber/rails/rspec.rb
@@ -3,7 +3,7 @@ require 'cucumber/rails/world'
 begin
   require 'rspec/rails/matchers'
 
-  [Cucumber::Rails::World, ActionController::Integration::Session].each do |klass|
+  [Cucumber::Rails::World, ActionDispatch::Integration::Session].each do |klass|
     klass.class_eval do
       include RSpec::Matchers
     end
@@ -12,7 +12,7 @@ rescue LoadError => try_rspec_1
   require 'spec/expectations'
   require 'spec/rails'
 
-  [Cucumber::Rails::World, ActionController::Integration::Session].each do |klass|
+  [Cucumber::Rails::World, ActionDispatch::Integration::Session].each do |klass|
     klass.class_eval do
       include Spec::Matchers
       include Spec::Rails::Matchers if defined?(Spec::Rails::Matchers)
Fix ActionController::Integration deprecations

ActionController::Integration has been deprecated in Rails <I> and
was removed in Rails <I>. User ActionDispatch::Integration instead.