[demo] fix: wrapper returning children on wallet disconnect

Envoy-VC · Dec 28, 2023 · 06f2ead · 06f2ead
1 parent c1dba46
commit 06f2ead
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/apps/demo/src/components/common/wrapper/index.tsx b/apps/demo/src/components/common/wrapper/index.tsx
@@ -1,16 +1,17 @@
 import React from 'react';
 import { useActiveAddress } from 'arweave-wallet-kit';
 import { useAtomicToolkit } from '~/stores';
-import { Button } from 'antd';
+import { useAddress } from '@thirdweb-dev/react';
 
 interface Props {
 	children: React.ReactNode;
 }
 
 const Wrapper = ({ children }: Props) => {
-	const { atomicToolkit, setAtomicToolkit } = useAtomicToolkit();
+	const { atomicToolkit } = useAtomicToolkit();
 	const activeAddress = useActiveAddress();
-	if (atomicToolkit) {
+	const ethAddress = useAddress();
+	if (atomicToolkit && (activeAddress || ethAddress)) {
 		return <>{children}</>;
 	} else {
 		return (