diff --git a/batcher-ui/src/contexts/wallet.tsx b/batcher-ui/src/contexts/wallet.tsx
index 0152c3a8..c752236e 100644
--- a/batcher-ui/src/contexts/wallet.tsx
+++ b/batcher-ui/src/contexts/wallet.tsx
@@ -1,9 +1,10 @@
 import { TezosToolkit } from '@taquito/taquito';
 import React, { createContext, useEffect, useReducer } from 'react';
 import { BeaconWallet } from '@taquito/beacon-wallet';
-import { AccountInfo, NetworkType } from '@airgap/beacon-sdk';
+import { AccountInfo } from '@airgap/beacon-sdk';
 import { useTezosToolkit } from './tezos-toolkit';
 import { getByKey, setByKey } from 'src/utils/local-storage';
+import { NetworkType } from '@airgap/beacon-types';
 
 type WalletState = {
   wallet: BeaconWallet | undefined;