Move local-mode server connectivity code to Chef::LocalMode

1 files changed, 3 insertions, 3 deletions

diff --git a/lib/chef/knife.rb b/lib/chef/knife.rb
index f8622975e8..4874f28cd6 100644
--- a/lib/chef/knife.rb
+++ b/lib/chef/knife.rb
@@ -25,6 +25,7 @@ require 'chef/mixin/convert_to_class_name'
 require 'chef/mixin/path_sanity'
 require 'chef/knife/core/subcommand_loader'
 require 'chef/knife/core/ui'
+require 'chef/local_mode'
 require 'chef/rest'
 require 'pp'
 
@@ -488,11 +489,11 @@ class Chef
         ui.error "You need to add a #run method to your knife command before you can use it"
       end
       enforce_path_sanity
-      Chef::Application.setup_server_connectivity
+      Chef::LocalMode.setup_server_connectivity
       begin
         run
       ensure
-        Chef::Application.destroy_server_connectivity
+        Chef::LocalMode.destroy_server_connectivity
       end
     rescue Exception => e
       raise if raise_exception || Chef::Config[:verbosity] == 2
@@ -625,4 +626,3 @@ class Chef
 
   end
 end
-