diff --git a/yarp/templates/template.rb b/yarp/templates/template.rb
index c661dd04e73f44..5880f11957bb9a 100755
--- a/yarp/templates/template.rb
+++ b/yarp/templates/template.rb
@@ -266,7 +266,6 @@ def template(name, write_to: nil)
       template = File.expand_path("../#{filepath}", __dir__)
 
       erb = read_template(template)
-      erb.filename = template
 
       non_ruby_heading = <<~HEADING
       /******************************************************************************/
@@ -305,9 +304,17 @@ def template(name, write_to: nil)
 
     def read_template(filepath)
       template = File.read(filepath, encoding: Encoding::UTF_8)
-      if ERB.instance_method(:initialize).parameters.assoc(:key) # Ruby 2.6+
+      erb = erb(template)
+      erb.filename = filepath
+      erb
+    end
+
+    if ERB.instance_method(:initialize).parameters.assoc(:key) # Ruby 2.6+
+      def erb(template)
         ERB.new(template, trim_mode: "-")
-      else
+      end
+    else
+      def erb
         ERB.new(template, nil, "-")
       end
     end