diff --git a/.env.example b/.env.example
index fe5b903..f796894 100644
--- a/.env.example
+++ b/.env.example
@@ -1,4 +1,4 @@
-APP_VERSION=0.7.0
+APP_VERSION=0.7.1
 
 NEXT_PUBLIC_PINATA_ENDPOINT_URL=https://api.pinata.cloud/pinning/pinFileToIPFS
 NEXT_PUBLIC_SG721_CODE_ID=2595
diff --git a/components/openEdition/OffChainMetadataUploadDetails.tsx b/components/openEdition/OffChainMetadataUploadDetails.tsx
index 3c785bb..184435e 100644
--- a/components/openEdition/OffChainMetadataUploadDetails.tsx
+++ b/components/openEdition/OffChainMetadataUploadDetails.tsx
@@ -294,7 +294,7 @@ export const OffChainMetadataUploadDetails = ({
             <div>
               <Tooltip
                 backgroundColor="bg-blue-500"
-                className="mb-2 ml-28"
+                className="mb-2 ml-4"
                 label="The token URI that points directly to the metadata file stored on IPFS."
                 placement="top"
               >
diff --git a/package.json b/package.json
index f35e212..003929a 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "stargaze-studio",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "workspaces": [
     "packages/*"
   ],
diff --git a/pages/collections/create.tsx b/pages/collections/create.tsx
index c732ef3..6a99323 100644
--- a/pages/collections/create.tsx
+++ b/pages/collections/create.tsx
@@ -147,8 +147,15 @@ const CollectionCreationPage: NextPage = () => {
             .then(() => {
               checkWhitelistDetails()
                 .then(() => {
-                  checkwalletBalance()
-                  setReadyToCreateVm(true)
+                  void checkwalletBalance()
+                    .then(() => {
+                      setReadyToCreateVm(true)
+                    })
+                    .catch((error) => {
+                      toast.error(`${error.message}`, { style: { maxWidth: 'none' } })
+                      addLogItem({ id: uid(), message: error.message, type: 'Error', timestamp: new Date() })
+                      setReadyToCreateVm(false)
+                    })
                 })
                 .catch((error) => {
                   if (String(error.message).includes('Insufficient wallet balance')) {
@@ -191,8 +198,15 @@ const CollectionCreationPage: NextPage = () => {
             .then(() => {
               checkWhitelistDetails()
                 .then(() => {
-                  checkwalletBalance()
-                  setReadyToCreateBm(true)
+                  void checkwalletBalance()
+                    .then(() => {
+                      setReadyToCreateBm(true)
+                    })
+                    .catch((error) => {
+                      toast.error(`${error.message}`, { style: { maxWidth: 'none' } })
+                      addLogItem({ id: uid(), message: error.message, type: 'Error', timestamp: new Date() })
+                      setReadyToCreateVm(false)
+                    })
                 })
                 .catch((error) => {
                   toast.error(`${error.message}`, { style: { maxWidth: 'none' } })
@@ -524,7 +538,11 @@ const CollectionCreationPage: NextPage = () => {
           whitelist,
         },
         collection_params: {
-          code_id: collectionDetails?.updatable ? SG721_UPDATABLE_CODE_ID : SG721_CODE_ID,
+          code_id: collectionDetails?.updatable
+            ? whitelistDetails?.whitelistType === 'flex'
+              ? SG721_CODE_ID
+              : SG721_UPDATABLE_CODE_ID
+            : SG721_CODE_ID,
           name: collectionDetails?.name,
           symbol: collectionDetails?.symbol,
           info: {
@@ -1104,40 +1122,41 @@ const CollectionCreationPage: NextPage = () => {
     }
   }
 
-  const checkwalletBalance = () => {
-    if (!wallet.initialized) throw new Error('Wallet not connected.')
-    if (minterType === 'vending' && whitelistDetails?.whitelistState === 'new' && whitelistDetails.memberLimit) {
-      const amountNeeded =
-        Math.ceil(Number(whitelistDetails.memberLimit) / 1000) * 100000000 +
-        (whitelistDetails.whitelistType === 'flex'
-          ? Number(vendingMinterFlexCreationFee)
-          : collectionDetails?.updatable
-          ? Number(vendingMinterUpdatableCreationFee)
-          : Number(vendingMinterCreationFee))
-      if (amountNeeded >= Number(wallet.balance[0].amount))
-        throw new Error(
-          `Insufficient wallet balance to instantiate the required contracts. Needed amount: ${(
-            amountNeeded / 1000000
-          ).toString()} STARS`,
-        )
-    } else {
-      const amountNeeded =
-        minterType === 'vending'
-          ? whitelistDetails?.whitelistState === 'existing' && whitelistDetails.whitelistType === 'flex'
+  const checkwalletBalance = async () => {
+    const walletBalance = await wallet.client?.getBalance(wallet.address, 'ustars').then((balance) => {
+      if (minterType === 'vending' && whitelistDetails?.whitelistState === 'new' && whitelistDetails.memberLimit) {
+        const amountNeeded =
+          Math.ceil(Number(whitelistDetails.memberLimit) / 1000) * 100000000 +
+          (whitelistDetails.whitelistType === 'flex'
             ? Number(vendingMinterFlexCreationFee)
             : collectionDetails?.updatable
             ? Number(vendingMinterUpdatableCreationFee)
-            : Number(vendingMinterCreationFee)
-          : collectionDetails?.updatable
-          ? Number(baseMinterUpdatableCreationFee)
-          : Number(baseMinterCreationFee)
-      if (amountNeeded >= Number(wallet.balance[0].amount))
-        throw new Error(
-          `Insufficient wallet balance to instantiate the required contracts. Needed amount: ${(
-            amountNeeded / 1000000
-          ).toString()} STARS`,
-        )
-    }
+            : Number(vendingMinterCreationFee))
+        if (amountNeeded >= Number(balance.amount))
+          throw new Error(
+            `Insufficient wallet balance to instantiate the required contracts. Needed amount: ${(
+              amountNeeded / 1000000
+            ).toString()} STARS`,
+          )
+      } else {
+        const amountNeeded =
+          minterType === 'vending'
+            ? whitelistDetails?.whitelistState === 'existing' && whitelistDetails.whitelistType === 'flex'
+              ? Number(vendingMinterFlexCreationFee)
+              : collectionDetails?.updatable
+              ? Number(vendingMinterUpdatableCreationFee)
+              : Number(vendingMinterCreationFee)
+            : collectionDetails?.updatable
+            ? Number(baseMinterUpdatableCreationFee)
+            : Number(baseMinterCreationFee)
+        if (amountNeeded >= Number(balance.amount))
+          throw new Error(
+            `Insufficient wallet balance to instantiate the required contracts. Needed amount: ${(
+              amountNeeded / 1000000
+            ).toString()} STARS`,
+          )
+      }
+    })
   }
 
   const syncCollections = useCallback(async () => {