diff --git a/editor/visual_script_editor.cpp b/editor/visual_script_editor.cpp
index bce082a..0417c96 100644
--- a/editor/visual_script_editor.cpp
+++ b/editor/visual_script_editor.cpp
@@ -4379,7 +4379,7 @@ void VisualScriptEditor::_notification(int p_what) {
 			func_input_scroll->add_theme_style_override(
 					"bg", get_theme_stylebox(SNAME("bg"), SNAME("Tree")));
 
-			Ref<Theme> tm = EditorNode::get_singleton()->get_theme_base()->get_theme();
+			Ref<Theme> tm = EditorNode::get_singleton()-> get_editor_theme();
 
 			bool dark_theme = tm->get_constant("dark_theme", "Editor");
 
diff --git a/visual_script.cpp b/visual_script.cpp
index 725e283..fac208a 100644
--- a/visual_script.cpp
+++ b/visual_script.cpp
@@ -2746,3 +2746,7 @@ VisualScriptLanguage::~VisualScriptLanguage() {
 }
 
 StringName VisualScript::get_global_name() const { return get_name(); }
+
+String VisualScript::get_class_icon_path() const {
+	return "icons/VisualScript.svg";
+}
diff --git a/visual_script.h b/visual_script.h
index 7192a27..ec46fe0 100644
--- a/visual_script.h
+++ b/visual_script.h
@@ -370,6 +370,7 @@ public:
 	virtual void set_source_code(const String &p_code) override;
 	virtual Error reload(bool p_keep_state = false) override;
 
+	virtual String get_class_icon_path() const;
 #ifdef TOOLS_ENABLED
 	virtual Vector<DocData::ClassDoc> get_documentation() const override {
 		Vector<DocData::ClassDoc> docs;