diff --git a/.vscode/extensions.json b/.vscode/extensions.json
index 37595818..67fa4e34 100644
--- a/.vscode/extensions.json
+++ b/.vscode/extensions.json
@@ -6,6 +6,8 @@
     "github.vscode-pull-request-github",
     "esbenp.prettier-vscode",
     "davidanson.vscode-markdownlint",
-    "yzhang.markdown-all-in-one"
+    "yzhang.markdown-all-in-one",
+    "bungcip.better-toml",
+    "redhat.vscode-yaml"
   ]
 }
diff --git a/src/config.go b/src/config.go
index e6a4e4a5..a559ef30 100644
--- a/src/config.go
+++ b/src/config.go
@@ -142,6 +142,11 @@ func exportConfig(configFile, format string) string {
 		prefix := "# yaml-language-server: $schema=https://raw.githubusercontent.com/JanDeDobbeleer/oh-my-posh/main/themes/schema.json\n\n"
 		content := buf.String()
 		return prefix + content
+
+	case config.Toml:
+		prefix := "#:schema https://raw.githubusercontent.com/JanDeDobbeleer/oh-my-posh/main/themes/schema.json\n\n"
+		content := buf.String()
+		return prefix + content
 	}
 
 	return buf.String()