lib/capistrano/puma.rb in capistrano3-puma-3.0.0 vs lib/capistrano/puma.rb in capistrano3-puma-3.0.1

@@ -28,9 +28,10 @@
       end.join("\n")
     end
 
 
     def template_puma(from, to, role)
+      @role = role
       file = [
           "lib/capistrano/templates/#{from}-#{role.hostname}-#{fetch(:stage)}.rb",
           "lib/capistrano/templates/#{from}-#{role.hostname}.rb",
           "lib/capistrano/templates/#{from}-#{fetch(:stage)}.rb",
           "lib/capistrano/templates/#{from}.rb.erb",
\ No newline at end of file