diff --git a/src/bitmessageqt/__init__.py b/src/bitmessageqt/__init__.py
index fa6d77ca..a4d8c62f 100644
--- a/src/bitmessageqt/__init__.py
+++ b/src/bitmessageqt/__init__.py
@@ -3295,8 +3295,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'):