lib/capistrano/capistrano_plugin_template.rb in capistrano-template-0.0.2 vs lib/capistrano/capistrano_plugin_template.rb in capistrano-template-0.0.3

@@ -1,5 +1,9 @@
 require 'capistrano/template'
+require 'sshkit/all'
 
-include Capistrano::Template::Helpers::DSL
+# don't pollute global namespace
+extend Capistrano::Template::Helpers::DSL
+
+SSHKit::Backend::Netssh.send(:include, Capistrano::Template::Helpers::DSL)
 
 import File.join(__dir__, 'template', 'tasks', 'template_defaults.rake')