diff --git a/drift-common b/drift-common
index 711027e..662f61a 160000
--- a/drift-common
+++ b/drift-common
@@ -1 +1 @@
-Subproject commit 711027e73dcb567e7235df675e21d3cc45c87649
+Subproject commit 662f61ab590bb987449aa8d54204bbe8dfdca591
diff --git a/superstake-ui/src/hooks/useSPLTokenBalance.tsx b/superstake-ui/src/hooks/useSPLTokenBalance.tsx
index ba4033a..7a29783 100644
--- a/superstake-ui/src/hooks/useSPLTokenBalance.tsx
+++ b/superstake-ui/src/hooks/useSPLTokenBalance.tsx
@@ -63,8 +63,8 @@ const useSPLTokenBalance = (
 
 			const targetAccount = await getTokenAccount(
 				connection,
-				tokenMintAddress,
-				wallet.publicKey.toString()
+				new PublicKey(tokenMintAddress),
+				wallet.publicKey
 			);
 
 			if (!targetAccount) {
@@ -104,9 +104,8 @@ const useSPLTokenBalance = (
 		} else {
 			(async () => {
 				const targetAddress = await getTokenAddress(
-					tokenMintAddress,
-					// @ts-ignore
-					wallet.publicKey.toString()
+					new PublicKey(tokenMintAddress),
+					wallet.publicKey
 				);
 
 				setTargetAccountAddress(targetAddress.toString());