fix: assign proper module name for roles

PolymathNetwork · Oct 9, 2019 · eed78e8 · eed78e8
1 parent 37c2e6d
commit eed78e8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/PolymathBase.ts b/src/PolymathBase.ts
@@ -671,7 +671,7 @@ export class PolymathBase extends PolymathAPI {
         SecurityTokenRole.Erc20DividendsOperator,
       ].includes(role)
     ) {
-      moduleName = ModuleName.GeneralTransferManager;
+      moduleName = ModuleName.ERC20DividendCheckpoint;
       permission =
         role === SecurityTokenRole.Erc20DividendsAdministrator ? Perm.Admin : Perm.Operator;
     } else if (
@@ -680,7 +680,7 @@ export class PolymathBase extends PolymathAPI {
         SecurityTokenRole.EtherDividendsOperator,
       ].includes(role)
     ) {
-      moduleName = ModuleName.ERC20DividendCheckpoint;
+      moduleName = ModuleName.EtherDividendCheckpoint;
       permission =
         role === SecurityTokenRole.EtherDividendsAdministrator ? Perm.Admin : Perm.Operator;
     } else {

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -401,6 +401,7 @@ export interface ProcedureArguments {
   [ProcedureType.ControllerTransfer]: ControllerTransferProcedureArgs;
   [ProcedureType.SetController]: SetControllerProcedureArgs;
   [ProcedureType.AssignSecurityTokenRole]: AssignSecurityTokenRoleProcedureArgs;
+  [ProcedureType.AssignStoRole]: AssignStoRoleProcedureArgs;
   [ProcedureType.ModifyShareholderData]: ModifyShareholderDataProcedureArgs;
   [ProcedureType.RevokeKyc]: RevokeKycProcedureArgs;
   [ProcedureType.UnnamedProcedure]: {};