diff --git a/chatdialog.h b/chatdialog.h
index 75e5d97..7c366e5 100644
--- a/chatdialog.h
+++ b/chatdialog.h
@@ -58,6 +58,7 @@
   void createActions();
   QString formatTime(time_t);
   void printTimeInCell(QTextTable *, time_t);
+  void disableTreeDisplay();
 
 private slots:
   void returnPressed();
@@ -70,6 +71,7 @@
   void sendLeave();
   void replot();
   void updateRosterList(QStringList);
+  void enableTreeDisplay();
 
   // icon related
   void iconActivated(QSystemTrayIcon::ActivationReason reason);