diff --git a/certbot/tests/error_handler_test.py b/certbot/tests/error_handler_test.py
index 0974fe92a..a81d4a2d1 100644
--- a/certbot/tests/error_handler_test.py
+++ b/certbot/tests/error_handler_test.py
@@ -3,12 +3,12 @@ import os
 import signal
 import sys
 import unittest
-from contextlib import contextmanager
+import contextlib
 
 import mock
 
 
-@contextmanager
+@contextlib.contextmanager
 def signal_receiver(signums):
     """Context manager to catch signals"""
     signals = []