diff --git a/package.json b/package.json
index 2b797e1..c5facd5 100644
--- a/package.json
+++ b/package.json
@@ -19,7 +19,7 @@
         "onCommand:ssfs.reconnect",
         "onCommand:ssfs.disconnect"
     ],
-    "main": "./out/src/extension",
+    "main": "./out/extension.js",
     "author": {
         "name": "Kelvin Schoofs",
         "email": "schoofs.kelvin@gmail.com"
diff --git a/tsconfig.json b/tsconfig.json
index 79b7085..e83dbf6 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -8,10 +8,14 @@
             "es6"
         ],
         "sourceMap": true,
-        "rootDir": "."
+        "rootDir": "src"
     },
+    "compileOnSave": true,
+    "include": [
+        "src"
+    ],
     "exclude": [
         "node_modules",
         ".vscode-test"
     ]
-}
+}
\ No newline at end of file