diff --git a/Lib/test/test_winreg.py b/Lib/test/test_winreg.py
index c1a3d18..753be74 100644
--- a/Lib/test/test_winreg.py
+++ b/Lib/test/test_winreg.py
@@ -242,11 +242,14 @@
             key = OpenKey(HKEY_CURRENT_USER, test_key_name)
             self.assertNotEqual(key.handle, 0)
 
-            self.assertRaises(NotImplementedError, DisableReflectionKey(key))
-            self.assertRaises(NotImplementedError, EnableReflectionKey(key))
-            self.assertRaises(NotImplementedError, QueryReflectionKey(key))
-            self.assertRaises(NotImplementedError,
-                              DeleteKeyEx(HKEY_CURRENT_USER, test_key_name))
+            with self.assertRaises(NotImplementedError):
+                DisableReflectionKey(key)
+            with self.assertRaises(NotImplementedError):
+                EnableReflectionKey(key)
+            with self.assertRaises(NotImplementedError):
+                QueryReflectionKey(key)
+            with self.assertRaises(NotImplementedError):
+                DeleteKeyEx(HKEY_CURRENT_USER, test_key_name)
         finally:
             DeleteKey(HKEY_CURRENT_USER, test_key_name)