2 files changed, 5 insertions, 5 deletions
diff --git a/tests/python_dependencies/impacket/smbserver.py b/tests/python_dependencies/impacket/smbserver.py
index c481b27ba..902d22f9c 100644
--- a/tests/python_dependencies/impacket/smbserver.py
+++ b/tests/python_dependencies/impacket/smbserver.py
@@ -26,9 +26,10 @@ import socket
 import time
 import datetime
 import struct
-try: # Python 3
+import sys
+if sys.version_info.major >= 3:
     import configparser
-except ImportError: # Python 2
+else:
     import ConfigParser as configparser
 import SocketServer
 import threading
@@ -38,7 +39,6 @@ import ntpath
 import os
 import fnmatch
 import errno
-import sys
 import random
 import shutil
 from binascii import hexlify
diff --git a/tests/smbserver.py b/tests/smbserver.py
index 4fcd34f2b..fbebe1015 100755
--- a/tests/smbserver.py
+++ b/tests/smbserver.py
@@ -28,9 +28,9 @@ import os
 import sys
 import logging
 import tempfile
-try: # Python 3
+if sys.version_info.major >= 3:
     import configparser
-except ImportError: # Python 2
+else:
     import ConfigParser as configparser
 
 # Import our curl test data helper