From f340132c392da81a2ee393288ebc7b1ebafc7ead Mon Sep 17 00:00:00 2001
From: Boldizsar Mezei <boldizsar@unchained.work>
Date: Wed, 4 Oct 2023 08:21:25 +0200
Subject: [PATCH] Point IOTA to iota

---
 .../src/services/wallet/wallet.service.ts     | 19 ++++++++-----------
 packages/functions/test/milestone.sync.ts     |  2 +-
 packages/interfaces/src/utils.ts              |  4 ++--
 3 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/packages/functions/src/services/wallet/wallet.service.ts b/packages/functions/src/services/wallet/wallet.service.ts
index 4079ce8caf..ba6f62c3f8 100644
--- a/packages/functions/src/services/wallet/wallet.service.ts
+++ b/packages/functions/src/services/wallet/wallet.service.ts
@@ -26,20 +26,17 @@ export interface SendToManyTargets {
   nativeTokens?: NativeToken[];
 }
 
-// TODO set IOTA to correct node
 const NODES = {
   [Network.SMR]: ['https://smr1.svrs.io/', 'https://smr3.svrs.io/'],
   [Network.RMS]: ['https://rms1.svrs.io/'],
-  // [Network.IOTA]: [
-  //   'https://us3.svrs.io/',
-  //   'https://us4.svrs.io/',
-  //   'https://hs5.svrs.io/',
-  //   'https://hs6.svrs.io/',
-  //   'https://chrysalis-nodes.iota.org',
-  // ],
-  // [Network.ATOI]: ['https://devnet.svrs.io/'],
-  [Network.IOTA]: ['https://smr1.svrs.io/', 'https://smr3.svrs.io/'],
-  [Network.ATOI]: ['https://rms1.svrs.io/'],
+  [Network.IOTA]: [
+    'https://us3.svrs.io/',
+    'https://us4.svrs.io/',
+    'https://hs5.svrs.io/',
+    'https://hs6.svrs.io/',
+    'https://chrysalis-nodes.iota.org',
+  ],
+  [Network.ATOI]: ['https://devnet.svrs.io/'],
 };
 
 const getClient = async (network: Network) => {
diff --git a/packages/functions/test/milestone.sync.ts b/packages/functions/test/milestone.sync.ts
index 40023e256d..e73c2ce74a 100644
--- a/packages/functions/test/milestone.sync.ts
+++ b/packages/functions/test/milestone.sync.ts
@@ -76,4 +76,4 @@ const addressInDb = async (addresses: string[]) => {
 };
 
 syncMilestones(COL.MILESTONE_RMS);
-//TODO syncMilestones(COL.MILESTONE_ATOI, Network.ATOI);
+syncMilestones(COL.MILESTONE_ATOI);
diff --git a/packages/interfaces/src/utils.ts b/packages/interfaces/src/utils.ts
index bea3fa9acc..6c76216767 100644
--- a/packages/interfaces/src/utils.ts
+++ b/packages/interfaces/src/utils.ts
@@ -18,9 +18,9 @@ export const generateRandomFileName = () =>
 export const getMilestoneCol = (network: Network) => {
   switch (network) {
     case Network.IOTA:
-      return COL.MILESTONE_SMR;
+      return COL.MILESTONE;
     case Network.ATOI:
-      return COL.MILESTONE_RMS; //TODO set ti back to ATOI
+      return COL.MILESTONE_ATOI;
     case Network.SMR:
       return COL.MILESTONE_SMR;
     case Network.RMS: