diff --git a/codegen/lib/graphql_java_gen.rb b/codegen/lib/graphql_java_gen.rb
index dd91187..fa2212b 100644
--- a/codegen/lib/graphql_java_gen.rb
+++ b/codegen/lib/graphql_java_gen.rb
@@ -47,7 +47,7 @@ class << self
     private
 
     def erb_for(template_filename)
-      erb = ERB.new(File.read(template_filename), nil, '-')
+      erb = ERB.new(File.read(template_filename), trim_mode: '-')
       erb.filename = template_filename
       erb
     end
@@ -58,7 +58,7 @@ def erb_for(template_filename)
 
   def erb_for_entity(template)
     template_filename = File.expand_path("../graphql_java_gen/templates/#{template}.erb", __FILE__)
-    erb = ERB.new(File.read(template_filename), nil, '-')
+    erb = ERB.new(File.read(template_filename), trim_mode: '-')
     erb.filename = template_filename
     erb
   end