diff --git a/visual_script_editor.cpp b/visual_script_editor.cpp
index 7e54891..47c0730 100644
--- a/visual_script_editor.cpp
+++ b/visual_script_editor.cpp
@@ -2040,7 +2040,7 @@ void VisualScriptEditor::set_edit_state(const Variant &p_state) {
 
 	Dictionary d = p_state;
 	if (d.has("function")) {
-		edited_func = p_state;
+		edited_func = d["function"];
 		selected = edited_func;
 	}