diff --git a/src/lib/components/workspace/Functions/FunctionEditor.svelte b/src/lib/components/workspace/Functions/FunctionEditor.svelte
index 3f85c9f2a1..4837f8d5f5 100644
--- a/src/lib/components/workspace/Functions/FunctionEditor.svelte
+++ b/src/lib/components/workspace/Functions/FunctionEditor.svelte
@@ -347,6 +347,7 @@ class Pipe:
 					<CodeEditor
 						bind:this={codeEditor}
 						value={content}
+						lang="python"
 						{boilerplate}
 						on:change={(e) => {
 							_content = e.detail.value;
diff --git a/src/lib/components/workspace/Tools/ToolkitEditor.svelte b/src/lib/components/workspace/Tools/ToolkitEditor.svelte
index 61413821dc..0340e677d2 100644
--- a/src/lib/components/workspace/Tools/ToolkitEditor.svelte
+++ b/src/lib/components/workspace/Tools/ToolkitEditor.svelte
@@ -236,6 +236,7 @@ class Tools:
 						bind:this={codeEditor}
 						value={content}
 						{boilerplate}
+						lang="python"
 						on:change={(e) => {
 							_content = e.detail.value;
 						}}