fix: sto module error issue

PolymathNetwork · Jan 7, 2020 · a1a3e53 · a1a3e53
1 parent 946321e
commit a1a3e53
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/src/procedures/FinalizeSto.ts b/src/procedures/FinalizeSto.ts
@@ -75,7 +75,10 @@ export class FinalizeSto extends Procedure<FinalizeStoProcedureArgs> {
       });
     }
 
-    const stoModuleErrorMessage = `STO ${stoAddress} is either archived or hasn't been launched`;
+    const stoModuleError = new PolymathError({
+      code: ErrorCode.ProcedureValidationError,
+      message: `STO ${stoAddress} is either archived or hasn't been launched`,
+    });
 
     let stoModule;
     let remainingTokens: BigNumber;
@@ -88,10 +91,7 @@ export class FinalizeSto extends Procedure<FinalizeStoProcedureArgs> {
         });
 
         if (!stoModule) {
-          throw new PolymathError({
-            code: ErrorCode.ProcedureValidationError,
-            message: stoModuleErrorMessage,
-          });
+          throw stoModuleError;
         }
 
         if (isCappedSTO_3_0_0(stoModule)) {
@@ -114,10 +114,7 @@ export class FinalizeSto extends Procedure<FinalizeStoProcedureArgs> {
         });
 
         if (!stoModule) {
-          throw new PolymathError({
-            code: ErrorCode.ProcedureValidationError,
-            message: stoModuleErrorMessage,
-          });
+          throw stoModuleError;
         }
 
         const { tokensSold, capPerTier } = await stoModule.getSTODetails();