diff --git a/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerDiagnosticsTreeView.cs b/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerDiagnosticsTreeView.cs
index f2d11676..a2537cdc 100644
--- a/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerDiagnosticsTreeView.cs
+++ b/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerDiagnosticsTreeView.cs
@@ -6,6 +6,11 @@
 using UnityEditor.IMGUI.Controls;
 using VContainer.Diagnostics;
 using VContainer.Unity;
+#if UNITY_6000_2_OR_NEWER
+using TreeView = UnityEditor.IMGUI.Controls.TreeView<int>;
+using TreeViewItem = UnityEditor.IMGUI.Controls.TreeViewItem<int>;
+using TreeViewState = UnityEditor.IMGUI.Controls.TreeViewState<int>;
+#endif
 
 namespace VContainer.Editor.Diagnostics
 {
diff --git a/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerInstanceTreeView.cs b/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerInstanceTreeView.cs
index d0dc68c4..496bc1ce 100644
--- a/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerInstanceTreeView.cs
+++ b/VContainer/Assets/VContainer/Editor/Diagnostics/VContainerInstanceTreeView.cs
@@ -5,6 +5,11 @@
 using UnityEditor.IMGUI.Controls;
 using UnityEngine;
 using VContainer.Diagnostics;
+#if UNITY_6000_2_OR_NEWER
+using TreeView = UnityEditor.IMGUI.Controls.TreeView<int>;
+using TreeViewItem = UnityEditor.IMGUI.Controls.TreeViewItem<int>;
+using TreeViewState = UnityEditor.IMGUI.Controls.TreeViewState<int>;
+#endif
 
 namespace VContainer.Editor.Diagnostics
 {