diff --git a/Editor/Editors/TriEditor.cs b/Editor/Editors/TriEditor.cs
index e49a35c..668e782 100644
--- a/Editor/Editors/TriEditor.cs
+++ b/Editor/Editors/TriEditor.cs
@@ -17,6 +17,12 @@ protected virtual void OnDisable()
             _core.Dispose();
         }
 
+
+        public override void OnInspectorGUI()
+        {
+            _core.OnInspectorGUI();
+        }
+
         public override VisualElement CreateInspectorGUI()
         {
             return _core.CreateVisualElement();
diff --git a/Editor/Editors/TriScriptedImporterEditor.cs b/Editor/Editors/TriScriptedImporterEditor.cs
index f5a977f..de4f90f 100644
--- a/Editor/Editors/TriScriptedImporterEditor.cs
+++ b/Editor/Editors/TriScriptedImporterEditor.cs
@@ -30,6 +30,13 @@ public override void OnDisable()
             base.OnDisable();
         }
 
+        public override void OnInspectorGUI()
+        {
+            _core.OnInspectorGUI();
+
+            ApplyRevertGUI();
+        }
+
         public override VisualElement CreateInspectorGUI()
         {
             var root = new VisualElement();