diff --git a/packages/dapp/src/SignMessage.tsx b/packages/dapp/src/SignMessage.tsx
index fb2592a..96addda 100644
--- a/packages/dapp/src/SignMessage.tsx
+++ b/packages/dapp/src/SignMessage.tsx
@@ -6,7 +6,6 @@ import { Key as KeyIcon, CheckCircle as CheckCircleIcon } from '@mui/icons-mater
 import { web3FromSource } from '@polkagate/extension-dapp';
 import { Divider, Grid, TextField, Typography } from '@mui/material';
 import { ApiPromise } from '@polkadot/api';
-import type { DeriveBalancesAll } from '@polkadot/api-derive/types';
 import { SignerPayloadRaw } from '@polkadot/types/types';
 import { verifySignature } from './util/verifySignature';
 
diff --git a/packages/snap/package.json b/packages/snap/package.json
index 8aae3ad..0a29a9a 100644
--- a/packages/snap/package.json
+++ b/packages/snap/package.json
@@ -6,7 +6,7 @@
     "type": "git",
     "url": "https://github.com/polkagate/polkamask.git"
   },
-  "license": "(MIT-0 OR Apache-2.0)",
+  "license": "Apache-2.0",
   "main": "./dist/bundle.js",
   "files": [
     "dist/",
@@ -28,7 +28,6 @@
   "dependencies": {
     "@metamask/key-tree": "latest",
     "@metamask/snaps-sdk": "latest",
-    "@metamask/snaps-ui": "latest",
     "@metamask/utils": "latest",
     "@polkadot/api": "^10.11.1",
     "@polkadot/apps-config": "latest",
@@ -48,8 +47,9 @@
     "@metamask/eslint-config-jest": "^10.0.0",
     "@metamask/eslint-config-nodejs": "^10.0.0",
     "@metamask/eslint-config-typescript": "^10.0.0",
-    "@metamask/snaps-cli": "^2.0.1",
+    "@metamask/snaps-cli": "^4.0.0",
     "@metamask/snaps-jest": "^4.0.1",
+    "@metamask/utils": "latest",
     "@typescript-eslint/eslint-plugin": "^5.33.0",
     "@typescript-eslint/parser": "^5.33.0",
     "eslint": "^8.21.0",
diff --git a/packages/snap/src/index.test.ts b/packages/snap/src/index.test.ts
index 92eab93..7230927 100644
--- a/packages/snap/src/index.test.ts
+++ b/packages/snap/src/index.test.ts
@@ -3,7 +3,7 @@
 /* eslint-disable jest/prefer-strict-equal */
 
 import { installSnap } from '@metamask/snaps-jest';
-import { copyable, divider, heading, panel, text } from '@metamask/snaps-ui';
+import { copyable, divider, heading, panel, text } from '@metamask/snaps-sdk';
 import { Json } from '@metamask/utils';
 import { hexToU8a, isHex } from '@polkadot/util';
 // import { expect } from '@jest/globals';
diff --git a/packages/snap/src/util/getFormatted.ts b/packages/snap/src/util/getFormatted.ts
index 0ae8040..20494ac 100644
--- a/packages/snap/src/util/getFormatted.ts
+++ b/packages/snap/src/util/getFormatted.ts
@@ -2,11 +2,11 @@ import { decodeAddress, encodeAddress } from '@polkadot/util-crypto';
 import { getChain } from '../chains';
 
 /**
- * To get the formatted address of an address
+ * To get the formatted address of an address.
  *
- * @param genesisHash - the genesisHash of the chain will be used to find an endpoint to use
- * @param address - the substrate format of an address
- * @returns - the formatted address
+ * @param genesisHash - The genesisHash of the chain will be used to find an endpoint to use.
+ * @param address - The substrate format of an address.
+ * @returns The formatted address.
  */
 export function getFormatted(genesisHash: string, address: string): string {
   const { prefix } = getChain(genesisHash as string);