diff --git a/plugin.py b/plugin.py
index 543cc0b..0cd54e4 100644
--- a/plugin.py
+++ b/plugin.py
@@ -416,6 +416,11 @@ def on_legend_click(self, current, column=0):
             ret = self.is_pgDistant(layer)
             if ret:
                 try:
+                    if not self.versioning: 
+                        out = self.selectDatabase()
+                        (pg_conn_info_out, conn_dict) = self.get_conn_from_settings(out)
+                        self.versioning = versioning.pgLocal(
+                            pg_conn_info_out, uri.schema(), self.get_conn_from_uri(uri))
                     rev = self.versioning.revision()
                     selection_type = 'working copy'
                     self.info.setText(uri.database()+' '+uri.schema()
@@ -438,6 +443,10 @@ def on_legend_click(self, current, column=0):
                     # check if it's a working copy
                     rev = 0
                     try:
+                        if not self.versioning:
+                            self.versioning = versioning.pgServer(
+                                self.pg_conn_info(), uri.schema())
+                        
                         rev = self.versioning.revision()
                         selection_type = 'working copy'
                         self.info.setText(uri.database()+' '+uri.schema()