Fix broken import

KlimaDAO · Sep 11, 2024 · 7aee235 · 7aee235
1 parent fecb513
commit 7aee235
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')