diff --git a/src/orbitClient.ts b/src/orbitClient.ts
index d92562aa..3440c230 100644
--- a/src/orbitClient.ts
+++ b/src/orbitClient.ts
@@ -1,4 +1,4 @@
-import { Client, PublicClient, PublicClientConfig, createPublicClient } from 'viem';
+import { PublicClient, PublicClientConfig, createPublicClient } from 'viem';
 import { rollupCreator } from './contracts';
 import { ParentChainId, validParentChainId } from './types/ParentChain';
 
@@ -7,7 +7,7 @@ export interface OrbitClient extends PublicClient {
   getValidChainId(): ParentChainId;
 }
 
-const validateClientChainId = (client: Client) => {
+const validateClientChainId = (client: PublicClient) => {
   const chainId = client.chain?.id;
   if (!validParentChainId(chainId)) {
     throw new Error('chainId is undefined');
@@ -19,7 +19,7 @@ export function createOrbitClient({ chain, transport }: PublicClientConfig): Orb
   return createPublicClient({
     chain,
     transport,
-  }).extend((client: Client) => ({
+  }).extend((client: PublicClient) => ({
     getRollupCreatorAddress: () => {
       const chainId = validateClientChainId(client);
       return rollupCreator.address[chainId];