🐛 fix risk steward import issue (#21)

* 🐛 fix issue * 🐛 avoid using a temp var
aave-dao · Dec 3, 2024 · a60518c · a60518c
1 parent 3a4a6e9
commit a60518c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/generator/templates/proposal.template.ts b/generator/templates/proposal.template.ts
@@ -6,10 +6,11 @@ import {prefixWithPragma} from '../utils/constants';
 export const proposalTemplate = (
   options: Options,
   poolConfig: PoolConfig,
-  pool: PoolIdentifier,
+  pool: PoolIdentifier
 ) => {
   const {title, author, discussion} = options;
   const chain = getPoolChain(pool);
+
   const folderName = generateFolderName(options);
   const contractName = generateContractName(options, pool);
 
@@ -23,10 +24,11 @@ export const proposalTemplate = (
   * @title ${title || 'TODO'}
   * @author ${author || 'TODO'}
   * - discussion: ${discussion || 'TODO'}
-  * - deploy-command: make run-script contract=src/contracts/updates/${folderName}/${contractName}.sol:${contractName} network=${getChainAlias(chain)} broadcast=false generate_diff=true
+  * - deploy-command: make run-script contract=src/contracts/updates/${folderName}/${contractName}.sol:${contractName} network=${getChainAlias(
+    chain
+  )} broadcast=false generate_diff=true
   */
- contract ${contractName} is ${`RiskStewards${chain}`
- } {
+ contract ${contractName} is ${`RiskStewards${chain === 'Base' ? 'BaseChain' : chain}`} {
   function name() public pure override returns (string memory) {
     return '${contractName}';
   }