From 0d3ba69d1abb947976218a175c91fccfd2b5e948 Mon Sep 17 00:00:00 2001
From: atticusofsparta <atticusofsparta@protonmail.com>
Date: Tue, 28 Jan 2025 12:36:46 -0600
Subject: [PATCH] fix(ant registry): custom cu url for ant registry

---
 src/state/actions/dispatchArIOInteraction.ts |  5 ++++-
 src/state/actions/dispatchArNSUpdate.ts      | 12 +++++++++++-
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/state/actions/dispatchArIOInteraction.ts b/src/state/actions/dispatchArIOInteraction.ts
index dfdb7f436..77e7aa8cd 100644
--- a/src/state/actions/dispatchArIOInteraction.ts
+++ b/src/state/actions/dispatchArIOInteraction.ts
@@ -83,7 +83,10 @@ export default async function dispatchArIOInteraction({
 
           const antRegistry = ANTRegistry.init({
             signer,
-            processId: ANT_REGISTRY_ID,
+            process: new AOProcess({
+              processId: ANT_REGISTRY_ID,
+              ao,
+            }),
           });
           await antRegistry
             .register({
diff --git a/src/state/actions/dispatchArNSUpdate.ts b/src/state/actions/dispatchArNSUpdate.ts
index 0349975ea..59f1c4e7c 100644
--- a/src/state/actions/dispatchArNSUpdate.ts
+++ b/src/state/actions/dispatchArNSUpdate.ts
@@ -1,5 +1,7 @@
 import {
   ANT,
+  ANTRegistry,
+  ANT_REGISTRY_ID,
   AOProcess,
   AoANTHandler,
   AoANTState,
@@ -144,7 +146,15 @@ export function dispatchArNSUpdate({
   });
 
   emitter
-    .fetchProcessesOwnedByWallet({ address: walletAddress.toString() })
+    .fetchProcessesOwnedByWallet({
+      address: walletAddress.toString(),
+      antRegistry: ANTRegistry.init({
+        process: new AOProcess({
+          processId: ANT_REGISTRY_ID,
+          ao,
+        }),
+      }),
+    })
     .catch((e) =>
       errorHandler('Error getting assets owned by wallet: ' + e.message),
     );