diff --git a/monkey/tests/unit_tests/common/test_di_container.py b/monkey/tests/unit_tests/common/test_di_container.py
index 20f492e43..857168f82 100644
--- a/monkey/tests/unit_tests/common/test_di_container.py
+++ b/monkey/tests/unit_tests/common/test_di_container.py
@@ -112,8 +112,7 @@ def test_register_mixed_instance_and_type(container):
     assert id(test_2.service_b) != id(test_3.service_b)
 
 
-def test_unregistered_type():
-    container = DIContainer()
+def test_unregistered_type(container):
     with pytest.raises(ValueError):
         container.resolve(TestClass1)