test: missed cleanup

ApeWorX · Aug 27, 2024 · e0fa133 · e0fa133
1 parent 1a1682f
commit e0fa133
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/functional/test_compilers.py b/tests/functional/test_compilers.py
@@ -191,8 +191,9 @@ def __init__(self):
             list(compilers.compile([new_contract_0, new_contract_1], project=project_with_contract))
 
     finally:
-        if mock_compiler.ext in compilers.__dict__.get("registered_compilers", {}):
-            del compilers.__dict__["registered_compilers"][mock_compiler.ext]
+        for ext in (mock_compiler.ext, second_mock_compiler.ext):
+            if ext in compilers.__dict__.get("registered_compilers", {}):
+                del compilers.__dict__["registered_compilers"][ext]
 
 
 def test_compile_source(compilers):