1 files changed, 6 insertions, 1 deletions

diff --git a/src/WorkspaceNameTool.hh b/src/WorkspaceNameTool.hh
index 0bdb528..bd034b7 100644
--- a/src/WorkspaceNameTool.hh
+++ b/src/WorkspaceNameTool.hh
@@ -27,6 +27,7 @@
 
 #include "FbTk/TextButton.hh"
 #include "FbTk/Observer.hh"
+#include "FbTk/Signal.hh"
 
 class BScreen;
 class ToolTheme;
@@ -35,7 +36,8 @@ namespace FbTk {
 template <class T> class ThemeProxy;
 }
 
-class WorkspaceNameTool: public ToolbarItem, public FbTk::Observer {
+class WorkspaceNameTool: public ToolbarItem, public FbTk::Observer,
+                         private FbTk::SignalTracker {
 public:
     WorkspaceNameTool(const FbTk::FbWindow &parent, FbTk::ThemeProxy<ToolTheme> &theme, BScreen &screen);
     virtual ~WorkspaceNameTool();
@@ -59,6 +61,9 @@ public:
     void parentMoved() { m_button.parentMoved(); }
 
 private:
+    /// Called when workspace changed on \c screen
+    void updateForScreen(BScreen &screen);
+
     void renderTheme(unsigned char alpha);
     void reRender();
     void updateSizing();