diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py
index 092e94f..8520401 100644
--- a/Lib/test/test_socket.py
+++ b/Lib/test/test_socket.py
@@ -2,6 +2,7 @@
 
 import unittest
 from test import support
+from unittest.case import _ExpectedFailure
 
 import errno
 import io
@@ -241,7 +242,7 @@
             raise TypeError("test_func must be a callable function")
         try:
             test_func()
-        except unittest._ExpectedFailure:
+        except _ExpectedFailure:
             # We deliberately ignore expected failures
             pass
         except BaseException as e: