1 files changed, 5 insertions, 4 deletions
diff --git a/modules/gdscript/editor/script_templates/VisualShaderNodeCustom/basic.gd b/modules/gdscript/editor/script_templates/VisualShaderNodeCustom/basic.gd
index 283a95d3b4..458e22dae4 100644
--- a/modules/gdscript/editor/script_templates/VisualShaderNodeCustom/basic.gd
+++ b/modules/gdscript/editor/script_templates/VisualShaderNodeCustom/basic.gd
@@ -1,6 +1,7 @@
 # meta-description: Visual shader's node plugin template
 
 @tool
+# Having a class name is required for a custom node.
 class_name VisualShaderNode_CLASS_
 extends _BASE_
 
@@ -17,7 +18,7 @@ func _get_description() -> String:
 	return ""
 
 
-func _get_return_icon_type() -> int:
+func _get_return_icon_type() -> PortType:
 	return PORT_TYPE_SCALAR
 
 
@@ -29,7 +30,7 @@ func _get_input_port_name(port: int) -> String:
 	return ""
 
 
-func _get_input_port_type(port: int) -> int:
+func _get_input_port_type(port: int) -> PortType:
 	return PORT_TYPE_SCALAR
 
 
@@ -41,10 +42,10 @@ func _get_output_port_name(port: int) -> String:
 	return "result"
 
 
-func _get_output_port_type(port: int) -> int:
+func _get_output_port_type(port: int) -> PortType:
 	return PORT_TYPE_SCALAR
 
 
 func _get_code(input_vars: Array[String], output_vars: Array[String],
-		mode: int, type: int) -> String:
+		mode: Shader.Mode, type: VisualShader.Type) -> String:
 	return output_vars[0] + " = 0.0;"