1 files changed, 3 insertions, 3 deletions
diff --git a/gui/designwidget.h b/gui/designwidget.h
index d6af83a0..628586f4 100644
--- a/gui/designwidget.h
+++ b/gui/designwidget.h
@@ -51,7 +51,7 @@ class DesignWidget : public QWidget
     void updateButtons();
     void addToHistory(QModelIndex item);
     std::vector<DecalXY> getDecals(ElementType type, IdString value);
-    void updateHighlightGroup(QList<LazyTreeItem *> item, int group);
+    void updateHighlightGroup(QList<TreeModel::Item *> item, int group);
   Q_SIGNALS:
     void info(std::string text);
     void selected(std::vector<DecalXY> decal, bool keep);
@@ -77,7 +77,7 @@ class DesignWidget : public QWidget
 
     QTreeView *treeView;
     QItemSelectionModel *selectionModel;
-    ContextTreeModel *treeModel;
+    TreeModel::Model *treeModel;
     QLineEdit *searchEdit;
     QtVariantPropertyManager *variantManager;
     QtVariantPropertyManager *readOnlyManager;
@@ -99,7 +99,7 @@ class DesignWidget : public QWidget
     QAction *actionClear;
 
     QColor highlightColors[8];
-    QMap<LazyTreeItem *, int> highlightSelected;
+    QMap<TreeModel::Item *, int> highlightSelected;
 
     QString currentSearch;
     QList<QModelIndex> currentSearchIndexes;