fix: expose ozVotesStorageProof strategy

snapshot-labs · Jan 30, 2024 · ae2d89e · ae2d89e
1 parent 9dc7be0
commit ae2d89e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/strategies/starknet/index.ts b/src/strategies/starknet/index.ts
@@ -4,6 +4,7 @@ import createErc20VotesStrategy from './erc20Votes';
 import { hexPadLeft } from '../../utils/encoding';
 import type { Strategy, NetworkConfig } from '../../types';
 import createEvmSlotValueStrategy from './evmSlotValue';
+import createOzVotesStorageProofStrategy from './ozVotesStorageProof';
 
 export function getStrategy(address: string, networkConfig: NetworkConfig): Strategy | null {
   const strategy = networkConfig.strategies[hexPadLeft(address)];
@@ -25,5 +26,9 @@ export function getStrategy(address: string, networkConfig: NetworkConfig): Stra
     return createEvmSlotValueStrategy(strategy.params);
   }
 
+  if (strategy.type === 'ozVotesStorageProof') {
+    return createOzVotesStorageProofStrategy(strategy.params);
+  }
+
   return null;
 }