Gemfile in beaker-puppet_install_helper-0.9.4 vs Gemfile in beaker-puppet_install_helper-0.9.6

@@ -2,10 +2,10 @@
 
 gemspec
 
 group :test do
   gem 'beaker', '= 2.52.0' if RUBY_VERSION <= '2.1.6'
-  gem 'beaker' if RUBY_VERSION > '2.1.6'
+  gem 'beaker', '~> 4' if RUBY_VERSION > '2.1.6'
   gem 'bundler', '~> 1.9'
   gem 'rake', '~> 10.0'
   gem 'rspec', '~> 3'
   gem 'rubocop', require: false
   gem 'rubocop-rspec', require: false