diff --git a/.github/workflows/gcr-publish.yml b/.github/workflows/gcr-publish.yml
index e13b471..279de53 100644
--- a/.github/workflows/gcr-publish.yml
+++ b/.github/workflows/gcr-publish.yml
@@ -5,7 +5,7 @@ name: Node.js Package
 
 on:
   push:
-    branches: [dev]
+    branches: [main]
   release:
     types: [created]
 
diff --git a/.github/workflows/npm-publish.yml b/.github/workflows/npm-publish.yml
index 648b278..5f4ad19 100644
--- a/.github/workflows/npm-publish.yml
+++ b/.github/workflows/npm-publish.yml
@@ -26,20 +26,15 @@ jobs:
       - uses: oven-sh/setup-bun@v2
       - run: bun install
       - run: bun run build
-      - run: |
+      - name: Check if version is already published
+        run: |
+          PACKAGE_VERSION=$(node -p "require('./package.json').version")
           PACKAGE_NAME=$(jq -r '.name' package.json)
-          PACKAGE_VERSION=$(jq -r '.version' package.json)
-
-          # Get the version from npm registry
-          REGISTRY_VERSION=$(npm show "$PACKAGE_NAME" version)
-
-          # Compare versions
-          if [ "$PACKAGE_VERSION" == "$REGISTRY_VERSION" ]; then
-            echo "Version $PACKAGE_VERSION already exists in the npm registry."
-            exit 0
-          else
-            echo "Version $PACKAGE_VERSION does not exist in the npm registry. Proceeding..."
-            npm publish --provenance --access public
-          fi
+          npm view $PACKAGE_NAME@$PACKAGE_VERSION
+        continue-on-error: true
+        id: check_version
+      - name: Publish
+        if: steps.check_version.outcome != 'success'
+        run: npm publish --provenance --access public
         env:
-          NODE_AUTH_TOKEN: ${{secrets.NPM_TOKEN}}
+          NODE_AUTH_TOKEN: ${{secrets.NPM_TOKEN}}
\ No newline at end of file
diff --git a/build.ts b/build.ts
index 6a7ba8f..b1bf3e2 100644
--- a/build.ts
+++ b/build.ts
@@ -1,11 +1,11 @@
 import { watch } from "fs"
 import { join } from "path"
-import { refreshTemplateWorkerExtension } from "@kksh/api/dev"
+import { refreshTemplateWorkerCommand } from "@kksh/api/dev"
 import { $ } from "bun"
 
 async function build() {
 	await $`bun build --minify --target=browser --outdir=./dist ./src/index.ts`
-	await refreshTemplateWorkerExtension()
+	await refreshTemplateWorkerCommand()
 }
 
 const srcDir = join(import.meta.dir, "src")
diff --git a/bun.lockb b/bun.lockb
index 4b19bd5..0712a67 100755
Binary files a/bun.lockb and b/bun.lockb differ
diff --git a/jsr.json b/jsr.json
index 591fdbf..70cee6e 100644
--- a/jsr.json
+++ b/jsr.json
@@ -1,6 +1,6 @@
 {
   "name": "@kunkun/kunkun-ext-ip-info",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "license": "MIT",
   "exports": "./mod.ts",
   "publish": {
diff --git a/package.json b/package.json
index 3dd03a7..398fc52 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
   "$schema": "https://schema.kunkun.sh/",
   "name": "kunkun-ext-ip-info",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "license": "MIT",
   "module": "index.ts",
   "repository": "https://github.com/kunkunsh/kunkun-ext-ip-info",
@@ -43,7 +43,7 @@
     "mod.ts"
   ],
   "dependencies": {
-    "@kksh/api": "^0.0.52"
+    "@kksh/api": "^0.1.1"
   },
   "devDependencies": {
     "@types/bun": "latest",
diff --git a/src/index.ts b/src/index.ts
index 1de75c6..bdb3ecc 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -9,8 +9,8 @@ import {
 	log,
 	toast,
 	ui,
-	WorkerExtension
-} from "@kksh/api/ui/worker"
+	TemplateUiCommand
+} from "@kksh/api/ui/template"
 import { boolean, number, object, parse, safeParse, string, type InferOutput } from "valibot"
 
 const IpApiJsonSchema = object({
@@ -68,7 +68,7 @@ function mapIpInfoToListItem(ip: IpListItem): List.Item {
 	})
 }
 
-class IpInfo extends WorkerExtension {
+class IpInfo extends TemplateUiCommand {
 	ip?: InferOutput<typeof IpApiJsonSchema>
 	listitems: List.Item[] = []