diff --git a/lib/rspec-puppet/tasks/release_test.rb b/lib/rspec-puppet/tasks/release_test.rb
index <HASH>..<HASH> 100644
--- a/lib/rspec-puppet/tasks/release_test.rb
+++ b/lib/rspec-puppet/tasks/release_test.rb
@@ -128,7 +128,7 @@ task :release_test do
   end
 end
 
-class GemfileRewrite < Parser::Rewriter
+class GemfileRewrite < Parser::TreeRewriter
   def on_send(node)
     _, method_name, *args = *node
Update release_test GemfileRewrite to use Parser::TreeRewriter