diff --git a/Lib/test/test_dict.py b/Lib/test/test_dict.py
index 9d013e3d2caa1c3..9b1f8d5b8db394b 100644
--- a/Lib/test/test_dict.py
+++ b/Lib/test/test_dict.py
@@ -121,18 +121,16 @@ def test_item_iterator_oom(self):
         import_helper.import_module('_testcapi')
         from test.support.script_helper import assert_python_ok
         code = """if 1:
-            import _testcapi
+            from test import support
             items = {1: 2, 3: 4}.items()
             ballast = [(i, i) for i in range(3000)]
             held = []
             for start in range(1, 5):
-                _testcapi.set_nomemory(start)
                 try:
-                    held.append(iter(items))
+                    with support.inject_memory_error_cm(start):
+                        held.append(iter(items))
                 except MemoryError:
                     pass
-                finally:
-                    _testcapi.remove_mem_hooks()
             """
         assert_python_ok('-c', code)