diff --git a/vyper/venom/passes/load_elimination.py b/vyper/venom/passes/load_elimination.py index c9d6f8c07a..feadc33e51 100644 --- a/vyper/venom/passes/load_elimination.py +++ b/vyper/venom/passes/load_elimination.py @@ -1,3 +1,4 @@ +from vyper.venom.effects import Effects from vyper.venom.analysis import DFGAnalysis, LivenessAnalysis, VarEquivalenceAnalysis from vyper.venom.passes.base_pass import IRPass @@ -27,11 +28,11 @@ def _process_bb(self, bb): memory = () for inst in bb.instructions: - if "memory" in inst.get_write_effects(): + if Effects.MEMORY in inst.get_write_effects(): memory = () - if "storage" in inst.get_write_effects(): + if Effects.STORAGE in inst.get_write_effects(): storage = () - if "transient" in inst.get_write_effects(): + if Effects.TRANSIENT in inst.get_write_effects(): transient = () if inst.opcode == "mstore":