diff --git a/Cargo.lock b/Cargo.lock
index b37af30f61..0d1656e08a 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -12682,9 +12682,9 @@ dependencies = [
 
 [[package]]
 name = "zed_toml"
-version = "0.0.2"
+version = "0.1.0"
 dependencies = [
- "zed_extension_api 0.0.5",
+ "zed_extension_api 0.0.6 (registry+https://github.com/rust-lang/crates.io-index)",
 ]
 
 [[package]]
diff --git a/extensions/toml/Cargo.toml b/extensions/toml/Cargo.toml
index deb7143d86..8b7f0a7f2f 100644
--- a/extensions/toml/Cargo.toml
+++ b/extensions/toml/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "zed_toml"
-version = "0.0.2"
+version = "0.1.0"
 edition = "2021"
 publish = false
 license = "Apache-2.0"
@@ -13,4 +13,4 @@ path = "src/toml.rs"
 crate-type = ["cdylib"]
 
 [dependencies]
-zed_extension_api = "0.0.5"
+zed_extension_api = "0.0.6"
diff --git a/extensions/toml/extension.toml b/extensions/toml/extension.toml
index b14319a36e..a09b83c01f 100644
--- a/extensions/toml/extension.toml
+++ b/extensions/toml/extension.toml
@@ -1,7 +1,7 @@
 id = "toml"
 name = "TOML"
 description = "TOML support."
-version = "0.0.2"
+version = "0.1.0"
 schema_version = 1
 authors = [
     "Max Brunsfeld <max@zed.dev>",
diff --git a/extensions/toml/src/toml.rs b/extensions/toml/src/toml.rs
index e629e1a3e8..2c355121f2 100644
--- a/extensions/toml/src/toml.rs
+++ b/extensions/toml/src/toml.rs
@@ -1,4 +1,5 @@
 use std::fs;
+use zed::LanguageServerId;
 use zed_extension_api::{self as zed, Result};
 
 struct TomlExtension {
@@ -6,7 +7,10 @@ struct TomlExtension {
 }
 
 impl TomlExtension {
-    fn language_server_binary_path(&mut self, config: zed::LanguageServerConfig) -> Result<String> {
+    fn language_server_binary_path(
+        &mut self,
+        language_server_id: &LanguageServerId,
+    ) -> Result<String> {
         if let Some(path) = &self.cached_binary_path {
             if fs::metadata(path).map_or(false, |stat| stat.is_file()) {
                 return Ok(path.clone());
@@ -14,7 +18,7 @@ impl TomlExtension {
         }
 
         zed::set_language_server_installation_status(
-            &config.name,
+            &language_server_id,
             &zed::LanguageServerInstallationStatus::CheckingForUpdate,
         );
         let release = zed::latest_github_release(
@@ -58,7 +62,7 @@ impl TomlExtension {
 
         if !fs::metadata(&binary_path).map_or(false, |stat| stat.is_file()) {
             zed::set_language_server_installation_status(
-                &config.name,
+                &language_server_id,
                 &zed::LanguageServerInstallationStatus::Downloading,
             );
 
@@ -98,11 +102,11 @@ impl zed::Extension for TomlExtension {
 
     fn language_server_command(
         &mut self,
-        config: zed::LanguageServerConfig,
+        language_server_id: &LanguageServerId,
         _worktree: &zed::Worktree,
     ) -> Result<zed::Command> {
         Ok(zed::Command {
-            command: self.language_server_binary_path(config)?,
+            command: self.language_server_binary_path(language_server_id)?,
             args: vec!["lsp".to_string(), "stdio".to_string()],
             env: Default::default(),
         })