Merge pull request #81 from 0xAurelius/hotfix/fix-middleware

Fix broken import
KlimaDAO · Sep 11, 2024 · 43f89f9 · 43f89f9
2 parents fecb513 + 7aee235
commit 43f89f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/staking_rewards/main.py b/src/staking_rewards/main.py
@@ -2,7 +2,7 @@
 import math
 import traceback
 
-from web3.middleware import geth_poa_middleware
+from web3.middleware import ExtraDataToPOAMiddleware
 from discord.ext import tasks
 
 from ..constants import SKLIMA_ADDRESS
@@ -18,7 +18,7 @@
 
 # Initialize web3
 web3 = get_polygon_web3()
-web3.middleware_onion.inject(geth_poa_middleware, layer=0)
+web3.middleware_onion.inject(ExtraDataToPOAMiddleware, layer=0)
 
 # Load ABIs
 SKLIMA_ABI = load_abi('sklima.json')