diff --git a/entrypoints/background.ts b/entrypoints/background.ts
index 92be5f8..44b04d5 100644
--- a/entrypoints/background.ts
+++ b/entrypoints/background.ts
@@ -49,18 +49,18 @@ async function handleTabUpdate(url: string) {
   }
 }
 
-chrome.tabs.onActivated.addListener(async activeInfo => {
-  const tab = await chrome.tabs.get(activeInfo.tabId)
+browser.tabs.onActivated.addListener(async activeInfo => {
+  const tab = await browser.tabs.get(activeInfo.tabId)
   if (tab.url) await handleTabUpdate(tab.url)
 })
 
-chrome.tabs.onUpdated.addListener(async (_tabId, changeInfo) => {
+browser.tabs.onUpdated.addListener(async (_tabId, changeInfo) => {
   if (changeInfo.url) await handleTabUpdate(changeInfo.url)
 })
 
 export default defineBackground({
   main() {
-    chrome.runtime.onMessage.addListener((request, _sender, sendResponse) => {
+    browser.runtime.onMessage.addListener((request: any, _sender, sendResponse) => {
       if (request.type === 'FETCH_SERVER_INFO') {
         ;(async () => {
           try {
@@ -99,6 +99,9 @@ export default defineBackground({
         })()
         return true
       }
+
+      sendResponse({ error: 'Unknown request type', data: null })
+      return true
     })
   },
 })
diff --git a/hooks/useTabData.ts b/hooks/useTabData.ts
index e32ef2e..5d32c8d 100644
--- a/hooks/useTabData.ts
+++ b/hooks/useTabData.ts
@@ -1,3 +1,8 @@
+import { useState, useEffect } from 'react';
+import browser from 'webextension-polyfill';
+import { isPrivateIP } from '@/utils';
+import { FetchServerInfoRequest, FetchServerInfoResponse, ServerData } from '@/utils/model';
+
 export function useTabData() {
   const [data, setData] = useState<ServerData | null>(null)
   const [loading, setLoading] = useState(true)
@@ -6,7 +11,7 @@ export function useTabData() {
   useEffect(() => {
     const fetchData = async () => {
       try {
-        const [tab] = await chrome.tabs.query({
+        const [tab] = await browser.tabs.query({
           active: true,
           currentWindow: true,
         })
@@ -43,7 +48,7 @@ export function useTabData() {
           })
         }
 
-        const response = await chrome.runtime.sendMessage({
+        const response = await browser.runtime.sendMessage<FetchServerInfoRequest, FetchServerInfoResponse>({
           type: 'FETCH_SERVER_INFO',
           hostname: hostname,
         })
diff --git a/package.json b/package.json
index 3bb7d84..63c95b6 100644
--- a/package.json
+++ b/package.json
@@ -2,7 +2,7 @@
   "name": "hostinfo",
   "description": "Receive information of a domain directly in the browser when browsing a website",
   "private": true,
-  "version": "1.1",
+  "version": "1.2",
   "type": "module",
   "scripts": {
     "dev": "wxt",
diff --git a/utils/icon.ts b/utils/icon.ts
index cab8e9d..912f32e 100644
--- a/utils/icon.ts
+++ b/utils/icon.ts
@@ -3,9 +3,9 @@ export async function updateIcon(countryCode: string | null) {
     countryCode?.match(/^[A-Z]{2}$/i)?.[0]?.toLowerCase() || 'unknown'
 
   const loadImageBitmap = async (code: string): Promise<ImageBitmap> => {
-    const url = chrome.runtime.getURL(`${code}.webp`)
+    const url = browser.runtime.getURL("/")
     try {
-      const response = await fetch(url)
+      const response = await fetch(url + `${code}.webp`)
       if (!response.ok) throw new Error('Flag not found')
       const blob = await response.blob()
       return await createImageBitmap(blob)
diff --git a/utils/model.ts b/utils/model.ts
index cfacee7..40c9f16 100644
--- a/utils/model.ts
+++ b/utils/model.ts
@@ -13,3 +13,13 @@ export interface DNSEntry {
   type: number
   data: string
 }
+
+export interface FetchServerInfoRequest {
+  type: 'FETCH_SERVER_INFO';
+  hostname: string;
+}
+
+export interface FetchServerInfoResponse {
+  error?: string;
+  data?: ServerData;
+}
diff --git a/wxt.config.ts b/wxt.config.ts
index 777330f..a01ac18 100644
--- a/wxt.config.ts
+++ b/wxt.config.ts
@@ -2,10 +2,10 @@ import { defineConfig } from 'wxt';
 import tailwindcss from "@tailwindcss/vite";
 
 export default defineConfig({
-  extensionApi: 'chrome',
+  extensionApi: 'webextension-polyfill',
   modules: ['@wxt-dev/module-react'],
   manifest: {
-    permissions: ['tabs', 'activeTab', 'webRequest', 'file://*', 'debugger'],
+    permissions: ['tabs', 'activeTab', 'webRequest'],
     host_permissions: ['https://ip.albert.lol/*', 'https://dns.google/*', 'https://flagcdn.com/*'],
     action: {
       default_title: 'Host Info',