diff --git a/registry/coder/modules/antigravity/main.test.ts b/registry/coder/modules/antigravity/main.test.ts
index 03adf117..a0d6e5c7 100644
--- a/registry/coder/modules/antigravity/main.test.ts
+++ b/registry/coder/modules/antigravity/main.test.ts
@@ -103,15 +103,18 @@ describe("antigravity", async () => {
       const mcp = JSON.stringify({
         servers: { demo: { url: "http://localhost:1234" } },
       });
+
       const state = await runTerraformApply(import.meta.dir, {
         agent_id: "foo",
         mcp,
       });
+      
       const script = findResourceInstance(
         state,
         "coder_script",
-        "antigravity_mcp",
+        "vscode-desktop-mcp", // from "vscode-desktop-core" module
       ).script;
+
       const resp = await execContainer(id, ["sh", "-c", script]);
       if (resp.exitCode !== 0) {
         console.log(resp.stdout);
diff --git a/registry/coder/modules/antigravity/main.tf b/registry/coder/modules/antigravity/main.tf
index 27c6166d..f54422e5 100644
--- a/registry/coder/modules/antigravity/main.tf
+++ b/registry/coder/modules/antigravity/main.tf
@@ -52,7 +52,7 @@ variable "display_name" {
 
 variable "mcp" {
   type        = string
-  description = "JSON-encoded string to configure MCP servers for Antigravity. When set, writes ~/.gemini/antigravity/mcp_config.json."
+  description = "JSON-encoded string to configure MCP servers for Antigravity. When set, writes $HOME/.gemini/antigravity/mcp_config.json."
   default     = ""
 }
 
@@ -60,45 +60,26 @@ data "coder_workspace" "me" {}
 
 data "coder_workspace_owner" "me" {}
 
-locals {
-  mcp_b64 = var.mcp != "" ? base64encode(var.mcp) : ""
-}
-
 module "vscode-desktop-core" {
-  source  = "registry.coder.com/coder/vscode-desktop-core/coder"
-  version = "1.0.1"
+  source = "git::https://github.com/coder/registry.git//registry/coder/modules/vscode-desktop-core?ref=phorcys/vscode-desktop-core-mcp"
 
   agent_id = var.agent_id
 
-  web_app_icon         = "/icon/antigravity.svg"
-  web_app_slug         = var.slug
-  web_app_display_name = var.display_name
-  web_app_order        = var.order
-  web_app_group        = var.group
+  coder_app_icon         = "/icon/antigravity.svg"
+  coder_app_slug         = var.slug
+  coder_app_display_name = var.display_name
+  coder_app_order        = var.order
+  coder_app_group        = var.group
 
   folder      = var.folder
   open_recent = var.open_recent
-  protocol    = "antigravity"
-}
+  mcp_config  = var.mcp
 
-resource "coder_script" "antigravity_mcp" {
-  count              = var.mcp != "" ? 1 : 0
-  agent_id           = var.agent_id
-  display_name       = "Antigravity MCP"
-  icon               = "/icon/antigravity.svg"
-  run_on_start       = true
-  start_blocks_login = false
-  script             = <<-EOT
-    #!/bin/sh
-    set -eu
-    mkdir -p "$HOME/.gemini/antigravity"
-    echo -n "${local.mcp_b64}" | base64 -d > "$HOME/.gemini/antigravity/mcp_config.json"
-    chmod 600 "$HOME/.gemini/antigravity/mcp_config.json"
-  EOT
+  protocol      = "antigravity"
+  config_folder = "$HOME/.gemini/antigravity"
 }
 
 output "antigravity_url" {
   value       = module.vscode-desktop-core.ide_uri
   description = "Antigravity IDE URL."
-}
-
+}
\ No newline at end of file