1 files changed, 12 insertions, 0 deletions
diff --git a/lib/puppet/network/client.rb b/lib/puppet/network/client.rb
index 13f1878fd..5603aeb43 100644
--- a/lib/puppet/network/client.rb
+++ b/lib/puppet/network/client.rb
@@ -36,6 +36,8 @@ class Puppet::Network::Client
 
     attr_accessor :schedule, :lastrun, :local, :stopping
 
+    attr_reader :driver
+
     # Set up subclass loading
     handle_subclasses :client, "puppet/network/client"
 
@@ -109,6 +111,16 @@ class Puppet::Network::Client
         end
     end
 
+    # Make sure we set the driver up when we read the cert in.
+    def read_cert
+        if super
+            @driver.cert_setup(self) if @driver.respond_to?(:cert_setup)
+            return true
+        else
+            return false
+        end
+    end
+
     # A wrapper method to run and then store the last run time
     def runnow
         if self.stopping