Fix check

apps/wallet-mobile/src/features/Settings/useCases/changeWalletSettings/Notifications/NotificationsDisplaySettings.tsx

@@ -2,27 +2,34 @@ import {isString, useAsyncStorage, useMutationWithInvalidations} from '@yoroi/co
 import {App} from '@yoroi/types'
 import {useQuery} from 'react-query'
 
-import {useSelectedWallet} from '../../../../WalletManager/common/hooks/useSelectedWallet'
+import {useWalletManager} from '../../../../WalletManager/context/WalletManagerProvider'
 
 const defaultNotificationsEnabled = true
 
 export const useNotificationDisplaySettings = () => {
   const storage = useAsyncStorage()
-  const selectedWallet = useSelectedWallet()
-  const selectedWalletId = selectedWallet.wallet.id
+  const walletManager = useWalletManager()
+  const selectedWalletId = walletManager.selected.wallet?.id
   const query = useQuery({
     queryKey: ['settings', selectedWalletId, 'notifications'],
-    queryFn: () => getNotificationDisplaySettings(storage, selectedWalletId),
+    queryFn: () => {
+      if (!isString(selectedWalletId)) return defaultNotificationsEnabled
+      return getNotificationDisplaySettings(storage, selectedWalletId)
+    },
+    enabled: isString(selectedWalletId),
   })
 
   return query.data ?? defaultNotificationsEnabled
 }
 
 export const useChangeNotificationDisplaySettings = () => {
   const storage = useAsyncStorage()
-  const selectedWallet = useSelectedWallet()
-  const selectedWalletId = selectedWallet.wallet.id
-  const mutationFn = (value: boolean) => changeNotificationDisplaySettings(storage, selectedWalletId, value)
+  const walletManager = useWalletManager()
+  const selectedWalletId = walletManager.selected.wallet?.id
+  const mutationFn = async (value: boolean) => {
+    if (!isString(selectedWalletId)) throw new Error('useChangeNotificationDisplaySettings: No wallet selected')
+    await changeNotificationDisplaySettings(storage, selectedWalletId, value)
+  }
   return useMutationWithInvalidations({
     mutationFn,
     invalidateQueries: [['settings', selectedWalletId, 'notifications']],