diff --git a/src/designer-plugins/treeview.cpp b/src/designer-plugins/treeview.cpp
index f9172f7..4773d31 100644
--- a/src/designer-plugins/treeview.cpp
+++ b/src/designer-plugins/treeview.cpp
@@ -8,7 +8,7 @@
 */
 
 #include "treeview.h"
-#include "app/qtkmainwindow.h"
+#include "qtkwidget.h"
 #include "ui_treeview.h"
 
 /*******************************************************************************
@@ -49,7 +49,7 @@ void Qtk::TreeView::updateView(const Qtk::Scene * scene)
 void Qtk::TreeView::itemFocus(QTreeWidgetItem * item, int column)
 {
   const QString & name = item->text(column);
-  auto scene = MainWindow::getMainWindow()->getQtkWidget()->getScene();
+  auto scene = QtkWidget::mWidgetManager.get_widget()->getScene();
   auto object = scene->getObject(name);
   // If the object is a mesh or model, focus the camera on it.
   if (object == Q_NULLPTR) {
@@ -70,5 +70,5 @@ void Qtk::TreeView::itemFocus(QTreeWidgetItem * item, int column)
   camera_transform.translate(0.0f, 0.0f, 3.0f);
 
   // Emit signal from qtk widget for new object focus. Triggers GUI updates.
-  emit MainWindow::getMainWindow() -> getQtkWidget()->objectFocusChanged(name);
+  emit QtkWidget::mWidgetManager.get_widget()->objectFocusChanged(name);
 }