lib/cloudstack-nagios/commands/snmpd_config.rb in cloudstack-nagios-0.18.2 vs lib/cloudstack-nagios/commands/snmpd_config.rb in cloudstack-nagios-0.18.3

@@ -1,10 +1,11 @@
-require 'sshkit/dsl'
+require 'sshkit'
 require 'socket'
 require 'timeout'
 
 class SnmpdConfig < CloudstackNagios::Base
+  include SSHKit::DSL
 
   desc "check [vms]", "check if snmpd is configured on virtual routers"
   def check(*vms)
     if vms.size == 0
       say 'Get a list of all routers from cloudstack..', :yellow
@@ -73,6 +74,6 @@
       ips
     end
 
   end
 
-end
\ No newline at end of file
+end