diff --git a/vscode/icons/icon-theme.json b/vscode/icons/icon-theme.json
deleted file mode 100644
index 6559bef..0000000
--- a/vscode/icons/icon-theme.json
+++ /dev/null
@@ -1,18 +0,0 @@
-{
-  "iconDefinitions": {
-    "simf": {
-      "iconPath": "./simf.svg"
-    },
-    "wit": {
-      "iconPath": "./wit.svg"
-    }
-  },
-  "fileExtensions": {
-    "simf": "simf",
-    "wit": "wit"
-  },
-  "languageIds": {
-    "simplicityhl": "simf",
-    "simplicityhl-witness": "wit"
-  }
-}
diff --git a/vscode/package.json b/vscode/package.json
index 7fde4a8..0bc207f 100644
--- a/vscode/package.json
+++ b/vscode/package.json
@@ -2,7 +2,7 @@
   "name": "simplicityhl",
   "displayName": "SimplicityHL Language Support",
   "description": "Syntax highlighting and autocompletion for SimplicityHL (Simfony) language",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "publisher": "Blockstream",
   "repository": {
     "type": "git",
@@ -28,6 +28,10 @@
         "extensions": [
           ".simf"
         ],
+        "icon": {
+          "dark": "./icons/simf.svg",
+          "light": "./icons/simf.svg"
+        },
         "configuration": "./language-configuration.json"
       },
       {
@@ -39,9 +43,12 @@
           "simplicityhl witness"
         ],
         "extensions": [
-          ".wit",
-          ".json"
+          ".wit"
         ],
+        "icon": {
+          "dark": "./icons/wit.svg",
+          "light": "./icons/wit.svg"
+        },
         "configuration": "./witness-configuration.json"
       }
     ],
@@ -203,13 +210,6 @@
         },
         "severity": "error"
       }
-    ],
-    "iconThemes": [
-      {
-        "id": "simplicityhl-icons",
-        "label": "SimplicityHL File Icons",
-        "path": "./icons/icon-theme.json"
-      }
     ]
   },
   "license": "MIT",