diff --git a/vs-code-extension/package.json b/vs-code-extension/package.json
index ac723a4..9c38a0d 100644
--- a/vs-code-extension/package.json
+++ b/vs-code-extension/package.json
@@ -15,7 +15,8 @@
 		"vscode": "^1.74.0"
 	},
 	"activationEvents": [
-		"onLanguage:plaintext"
+		"onLanguage:plaintext",
+		"onLanguage:yaml"
 	],
 	"main": "./out/extension",
 	"scripts": {
diff --git a/vs-code-extension/src/extension.ts b/vs-code-extension/src/extension.ts
index 38f1484..1035343 100644
--- a/vs-code-extension/src/extension.ts
+++ b/vs-code-extension/src/extension.ts
@@ -18,7 +18,13 @@ export function activate(context: ExtensionContext) {
 			{
 				scheme: 'file',
 				language: 'plaintext',
-				pattern: "**/sshconfig",
+				pattern: "**/{config,sshconfig,sshd_config,sshdconfig,fstab,hosts,aliases}",
+			},
+			// Some configs seem to be incorrectly detected as yaml
+			{
+				scheme: 'file',
+				language: 'yaml',
+				pattern: "**/{config,sshconfig,sshd_config,sshdconfig,fstab,hosts,aliases}",
 			},
 		]
 	};