diff --git a/src/bitmessageqt/__init__.py b/src/bitmessageqt/__init__.py
index 7a9e0e30..a4e654bc 100644
--- a/src/bitmessageqt/__init__.py
+++ b/src/bitmessageqt/__init__.py
@@ -3293,8 +3293,8 @@ class MyForm(QtGui.QMainWindow):
                 return False
 
     def getCurrentFolder(self):
-        #treeWidget = self.getCurrentTreeWidget()
-        treeWidget = self.ui.treeWidgetYourIdentities
+        treeWidget = self.getCurrentTreeWidget()
+        #treeWidget = self.ui.treeWidgetYourIdentities
         if treeWidget:
             currentItem = treeWidget.currentItem()
             if currentItem and hasattr(currentItem, 'folderName'):