fix: v3 config engine rates import generator

bgd-labs · Oct 25, 2023 · 60aa9c8 · 60aa9c8
1 parent 1e69d05
commit 60aa9c8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/generator/utils/importsResolver.ts b/generator/utils/importsResolver.ts
@@ -10,7 +10,7 @@ const GovernanceImports = [
 ] as const;
 
 /**
- * @dev matches the code fro known address book imports and generates an import statment satisfying the used libraries
+ * @dev matches the code fro known address book imports and generates an import statement satisfying the used libraries
  * @param code
  * @returns
  */
@@ -83,8 +83,8 @@ export function prefixWithImports(code: string) {
   if (findMatch(code, 'IAaveV3ConfigEngine')) {
     imports += `import {IAaveV3ConfigEngine} from 'aave-helpers/v3-config-engine/IAaveV3ConfigEngine.sol';\n`;
   }
-  if (findMatch(code, 'IV3RateStrategyFactory')) {
-    imports += `import {IV3RateStrategyFactory} from 'aave-helpers/v3-config-engine/IV3RateStrategyFactory.sol';\n`;
+  if (findMatch(code, 'Rates')) {
+    imports += `import {IV3RateStrategyFactory as Rates} from 'aave-helpers/v3-config-engine/IV3RateStrategyFactory.sol';\n`;
   }
   // v2 config engine imports
   if (findMatch(code, 'IAaveV2ConfigEngine')) {