diff --git a/apps/staking/src/components/WalletTester/index.tsx b/apps/staking/src/components/WalletTester/index.tsx
index 63574a20c2..00e506def6 100644
--- a/apps/staking/src/components/WalletTester/index.tsx
+++ b/apps/staking/src/components/WalletTester/index.tsx
@@ -21,8 +21,10 @@ import {
StateType as UseAsyncStateType,
} from "../../hooks/use-async";
import { useData, StateType as UseDataStateType } from "../../hooks/use-data";
+import { useNetwork } from "../../hooks/use-network";
import { useToast } from "../../hooks/use-toast";
import { Button } from "../Button";
+import { Switch } from "../Switch";
const MAX_TEST_RETRIES = 10;
@@ -65,13 +67,14 @@ const ConnectWallet = ({ isLoading }: { isLoading?: boolean | undefined }) => {
const showModal = useCallback(() => {
modal.setVisible(true);
}, [modal]);
+ const { isMainnet, toggleMainnet } = useNetwork();
return (
<>