From cbf0bf1010c4611565a377e13e83a518f26b9156 Mon Sep 17 00:00:00 2001
From: Andrew Murray <radarhere@users.noreply.github.com>
Date: Fri, 27 Mar 2020 22:15:04 +1100
Subject: [PATCH] Fixed restoring original state

---
 Tests/test_file_wmf.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Tests/test_file_wmf.py b/Tests/test_file_wmf.py
index 71b6012c3..03444eb9d 100644
--- a/Tests/test_file_wmf.py
+++ b/Tests/test_file_wmf.py
@@ -35,6 +35,7 @@ def test_register_handler(tmp_path):
             self.methodCalled = True
 
     handler = TestHandler()
+    original_handler = WmfImagePlugin._handler
     WmfImagePlugin.register_handler(handler)
 
     im = hopper()
@@ -43,7 +44,7 @@ def test_register_handler(tmp_path):
     assert handler.methodCalled
 
     # Restore the state before this test
-    WmfImagePlugin.register_handler(None)
+    WmfImagePlugin.register_handler(original_handler)
 
 
 def test_load_dpi_rounding():