lib/serverspec/setup.rb in serverspec-0.5.4 vs lib/serverspec/setup.rb in serverspec-0.5.5

@@ -122,11 +122,11 @@
       c.host  = host
       options = Net::SSH::Config.for(c.host)
       user    = options[:user] || Etc.getlogin
       ### include vagrant conf ###
       c.ssh   = Net::SSH.start(c.host, user, options)
-      c.os    = backend(Serverspec::Commands::Base).check_os
+      c.os    = backend.check_os
     end
   end")
             if @vagrant
               content.gsub!(/### include vagrant conf ###/,"
       config = `vagrant ssh-config --host \#{host}`
@@ -147,10 +147,10 @@
             else
               content.gsub!(/### include vagrant conf ###/,'')
             end
             when 'Exec'
               content.gsub!(/### include backend conf ###/, "c.before :all do
-    c.os = backend(Serverspec::Commands::Base).check_os
+    c.os = backend.check_os
   end")
             when 'Puppet'
               content.gsub!(/### include requirements ###/, "require 'puppet'\nrequire 'serverspec/backend/puppet'
 ")
           end