fix: fix how we get owned security tokens

PolymathNetwork · Feb 20, 2020 · d5dcb48 · d5dcb48
1 parent 454d285
commit d5dcb48
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/Polymath.ts b/src/Polymath.ts
@@ -295,19 +295,21 @@ export class Polymath {
       walletAddress = await currentWallet.address();
     }
 
-    const [ownedTickers, delegatedAddresses] = await Promise.all([
-      contractWrappers.securityTokenRegistry.getTickersByOwner({ owner: walletAddress }),
+    const [ownedTokens, delegatedTokens] = await Promise.all([
+      contractWrappers.securityTokenRegistry.getTokensByOwner({ owner: walletAddress }),
       contractWrappers.securityTokenRegistry.getTokensByDelegate(walletAddress),
     ]);
 
-    const delegateTickers = await P.map(delegatedAddresses, async address => {
+    const allTokens = union(ownedTokens, delegatedTokens);
+
+    const allTickers = await P.map(allTokens, async address => {
       const details = await contractWrappers.securityTokenRegistry.getSecurityTokenData({
         securityTokenAddress: address,
       });
       return details.ticker;
     });
 
-    return union(ownedTickers, delegateTickers);
+    return allTickers;
   };
 
   /**